34 lines
969 B
Python
34 lines
969 B
Python
|
|
# -*- 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
|