Files
HOSPITAL_CCTV/hpe_classification/config.py

34 lines
969 B
Python
Raw Normal View History

2026-02-25 15:05:58 +09:00
# -*- coding: utf-8 -*-
"""
@file : config.py
@author: yunikim
@license: A2TEC & DAOOLDNS
@brief:
@section Modify History
- 2023-11-21 yunikim base
"""
import logging
import project_config
from config_loader import CFG
_hpe = CFG.get('hpe', {})
DEFAULT_YOLO_VERSION = 8
FALLDOWN_TILT_RATIO = _hpe.get('falldown_tilt_ratio', 0.80)
FALLDOWN_TILT_ANGLE = _hpe.get('falldown_tilt_angle', 15)
BODY_TILT_RATIO = _hpe.get('body_tilt_ratio', 0.30)
CROSS_ARM_RATIO_THRESHOLD = _hpe.get('cross_arm_ratio_threshold', 0.10)
CROSS_ARM_ANGLE_THRESHOLD = tuple(_hpe.get('cross_arm_angle_threshold', [15.0, 165.0]))
if project_config.ADD_CROSS_ARM:
ARM_ANGLE_THRESHOLD = tuple(_hpe.get('arm_angle_threshold_cross', [80.0, 135.0]))
else:
ARM_ANGLE_THRESHOLD = tuple(_hpe.get('arm_angle_threshold_default', [150.0, 195.0]))
ELBOW_ANGLE_THRESHOLD = tuple(_hpe.get('elbow_angle_threshold', [150.0, 185.0]))
DEFAULT_LOG_LEVEL = logging.INFO