You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
class Config:
|
|
# ===== Камера =====
|
|
CAMERA_ID = 0 # только для 'web'
|
|
CAMERA_TYPE = 'oak' # 'web' или 'oak'
|
|
MIRROR_CAMERA = True
|
|
|
|
# ===== Скелет =====
|
|
POSE_DETECTOR = 'mediapipe' # 'oak' или 'mediapipe'
|
|
OAK_MODEL_TYPE = 'efficienthrnet2'
|
|
OAK_DETECTION_THRESHOLD = 0.1
|
|
OAK_SHAVES = 6
|
|
|
|
# ===== Преобразование позы в скорости =====
|
|
ARM_CONTROL = {
|
|
'linear_arm': 'right',
|
|
'angular_arm': 'left',
|
|
'max_speed_linear': 1.0,
|
|
'max_speed_angular': 1.0,
|
|
'dead_zone': 0.2,
|
|
'debug': False
|
|
}
|
|
|
|
# ===== Статические жесты (dome, cross) =====
|
|
SPECIAL_GESTURE_MODE = 'ml' # 'ml' или 'geometric'
|
|
ML_GESTURE_MODEL = '/home/ubuntu/sirius/models/rf/special_gestures_rf.pkl'
|
|
ML_GESTURE_CLASSES = ['dome', 'cross', 'none']
|
|
|
|
|
|
# ===== Динамические жесты =====
|
|
DYNAMIC_GESTURE = {
|
|
'enabled': True,
|
|
'model_path': '/home/ubuntu/sirius/models/lstm/dynamic_model.h5',
|
|
'classes_path': '/home/ubuntu/sirius/models/lstm/dynamic_model_classes.pkl',
|
|
'window_size': 14,
|
|
'threshold': 0.8
|
|
}
|
|
|
|
# ===== Симулятор робота =====
|
|
MAP_WIDTH = 800
|
|
MAP_HEIGHT = 600
|
|
BORDER_THICKNESS = 10
|
|
MAP_OBSTACLES = [
|
|
(200, 150, 100, 200),
|
|
(500, 300, 150, 50),
|
|
]
|
|
START_POS = (100, 100)
|
|
FINISH_POS = (700, 500)
|
|
ROBOT_RADIUS = 20
|
|
ROBOT_IMAGE_PATH = 'robot.png' # или None
|
|
|
|
INITIAL_HP = 100
|
|
INITIAL_FIRE_CHARGE = 4
|
|
COLLISION_DAMAGE = 10
|
|
|
|
# Конус перед роботом
|
|
EXTINGUISH_RADIUS = 50
|
|
EXTINGUISH_ANGLE = 60
|
|
|
|
NUM_FIRES = 4
|
|
FIRE_RADIUS = 15
|
|
SCORE_PER_FIRE = 10
|
|
|
|
ROBOT_MODE = 'simulator' # 'simulator' или 'debug'
|