88 lines
3.4 KiB
YAML
88 lines
3.4 KiB
YAML
|
# ===================================================
|
||
|
|
# KEPCO DEMO 설정 파일
|
||
|
|
# 이 파일을 수정하여 프로그램 동작을 변경할 수 있습니다.
|
||
|
|
# ===================================================
|
||
|
|
|
||
|
|
# --- 모델 설정 ---
|
||
|
|
pt_type: "dev" # "dev" 또는 "demo"
|
||
|
|
use_hpe_person: true # Human Pose Estimation 사용 여부
|
||
|
|
use_helmet_model: true # 헬멧 전용 모델 사용 여부
|
||
|
|
|
||
|
|
# --- 라벨 설정 ---
|
||
|
|
view_conf_score: false # confidence score 표시 여부
|
||
|
|
show_gloves: true # glove on/off 라벨링 여부
|
||
|
|
label_all_white: true # bounding box 색상 전부 흰색 여부
|
||
|
|
use_hpe_frame_check: false # n개 프레임 체크 후 HPE 위험 판단
|
||
|
|
|
||
|
|
# --- 디버그 ---
|
||
|
|
add_cross_arm: false # cross arm 디버그 모드
|
||
|
|
|
||
|
|
# --- 소스 설정 ---
|
||
|
|
# RTSP 카메라 프리셋 (source에서 프리셋 이름으로 사용 가능)
|
||
|
|
cameras:
|
||
|
|
MG: "rtsp://admin:admin1263!@10.20.10.99:28554/onvif/media?profile=Profile1"
|
||
|
|
LF: "rtsp://admin:!1q2w3e4r@10.20.10.100:50554/profile1/media.smp"
|
||
|
|
FC: "rtsp://daool:Ekdnfeldpsdptm1@192.168.200.101/axis-media/media.amp"
|
||
|
|
MG_74_LTE: "rtsp://admin:admin1263!@223.171.48.74:28554/trackID=1"
|
||
|
|
FERMAT: "rtsp://192.168.200.233:50399/wd"
|
||
|
|
|
||
|
|
source: "FC" # 카메라 프리셋 이름 또는 파일/RTSP 경로
|
||
|
|
save_path: "./250630_result"
|
||
|
|
|
||
|
|
# --- 모델 가중치 파일 ---
|
||
|
|
weights:
|
||
|
|
pose: "yolov8l-pose.pt"
|
||
|
|
helmet: "yolov8_dev1_97.pt"
|
||
|
|
# pt_type별 OD 모델
|
||
|
|
od_dev: "yolov11m_dev1_8.pt"
|
||
|
|
od_demo: "yolov8_dev1_66.pt"
|
||
|
|
|
||
|
|
# --- 모델 파라미터 ---
|
||
|
|
model_confidence: 0.5
|
||
|
|
model_image_size: 640
|
||
|
|
kpt_min_confidence: 0.5
|
||
|
|
|
||
|
|
# --- HPE 임계값 ---
|
||
|
|
hpe:
|
||
|
|
falldown_tilt_ratio: 0.80
|
||
|
|
falldown_tilt_angle: 15
|
||
|
|
body_tilt_ratio: 0.30
|
||
|
|
cross_arm_ratio_threshold: 0.10
|
||
|
|
cross_arm_angle_threshold: [15.0, 165.0]
|
||
|
|
arm_angle_threshold_cross: [80.0, 135.0] # add_cross_arm: true 일 때
|
||
|
|
arm_angle_threshold_default: [150.0, 195.0] # add_cross_arm: false 일 때
|
||
|
|
elbow_angle_threshold: [150.0, 185.0]
|
||
|
|
hpe_frame_check_max_count: 3
|
||
|
|
|
||
|
|
# --- 디스플레이 설정 ---
|
||
|
|
display:
|
||
|
|
turn_on: false # 화면 표시 여부
|
||
|
|
border_thickness: 40
|
||
|
|
border_thickness_half: 20
|
||
|
|
normal_thickness: 2
|
||
|
|
warning_thickness: 4
|
||
|
|
hpe_thickness_ratio: 1
|
||
|
|
text_size: 1
|
||
|
|
text_thickness: 0
|
||
|
|
ppe_union_min_percent: 0.9
|
||
|
|
loadstreams_img_buffer: 5
|
||
|
|
fhd_resolution: [1920, 1080]
|
||
|
|
|
||
|
|
# --- CCTV 관제 설정 ---
|
||
|
|
cctv:
|
||
|
|
switch_interval: 5.0
|
||
|
|
sources:
|
||
|
|
# - {src: 0, name: "Realtek_Webcam_0"}
|
||
|
|
# - {src: "rtsp://daool:Ekdnfeldpsdptm1@192.168.200.101/axis-media/media.amp", name: "ipcam_1"}
|
||
|
|
# - {src: "rtsp://daool:Ekdnfeldpsdptm1@192.168.200.102/axis-media/media.amp", name: "ipcam_2"}
|
||
|
|
- {src: "rtsp://192.168.200.231:50199/wd", name: "dev1_stream"}
|
||
|
|
- {src: "rtsp://192.168.200.232:50299/wd", name: "dev2_stream"}
|
||
|
|
# - {src: "rtsp://192.168.200.236:8554/videodevice", name: "boss_webcam"}
|
||
|
|
# - {src: "rtsp://192.168.200.236:8554/1.mp4", name: "boss_stream1"}
|
||
|
|
# - {src: "rtsp://192.168.200.214/1.mp4", name: "sgm_stream1"}
|
||
|
|
# - {src: "rtsp://192.168.200.214/2.mp4", name: "sgm_stream2"}
|
||
|
|
# - {src: "rtsp://192.168.200.111/1.mp4", name: "hp_stream1"}
|
||
|
|
# - {src: "rtsp://192.168.200.111/2.mp4", name: "hp_stream2"}
|
||
|
|
# - {src: "rtsp://192.168.200.215/1.mp4", name: "msk_stream1"}
|
||
|
|
# - {src: "rtsp://192.168.200.215/2.mp4", name: "msk_stream2"}
|