18 lines
540 B
Python
18 lines
540 B
Python
|
# === config.yaml에서 설정값 로드 ===
|
||
|
|
from config_loader import CFG
|
||
|
|
|
||
|
|
PT_TYPE = CFG.get('pt_type', 'dev')
|
||
|
|
|
||
|
|
# === use model ===
|
||
|
|
USE_HPE_PERSON = CFG.get('use_hpe_person', True)
|
||
|
|
USE_HELMET_MODEL = CFG.get('use_helmet_model', True)
|
||
|
|
|
||
|
|
# === label ===
|
||
|
|
VIEW_CONF_SCORE = CFG.get('view_conf_score', False)
|
||
|
|
SHOW_GLOVES = CFG.get('show_gloves', True)
|
||
|
|
LABEL_ALL_WHITE = CFG.get('label_all_white', True)
|
||
|
|
USE_HPE_FRAME_CHECK = CFG.get('use_hpe_frame_check', False)
|
||
|
|
|
||
|
|
# === debug ===
|
||
|
|
ADD_CROSS_ARM = CFG.get('add_cross_arm', False)
|