# YOLOX configurations
YOLOX_CONFIG_FILE = "modules/yolox/config/yolox_x_8x8_300e_coco-EARS-white.py"
YOLOX_CHECKPOINT_FILE = "models/yolox/yolox_x_White_only.pth"
DEVICE = "cuda:0"

# Colors for different models (B,G,R format)
CONV_COLOR = (27, 80, 19)  # 純粋な緑
XGBOOST_COLOR = (0, 165, 255)  # オレンジ
LIGHTGBM_COLOR = (255, 0, 255)  # マゼンタ
EARSNET_COLOR = (139, 0, 0)  # ダークブルー
CATBOOST_COLOR = (0, 0, 255)  # 赤
NGBOOST_COLOR = (255, 255, 0)  # シアン

# Model execution settings
CONV_ENABLED = True
XGBOOST_ENABLED = True
LIGHTGBM_ENABLED = True
CATBOOST_ENABLED = False
NGBOOST_ENABLED = False
NORMALIZE_ENABLED = True
POSENET_ENABLED = True
RTMPOSE_ENABLED = False
MOBILENETV1SSD_ENABLED = True
YOLOX_ENABLED = False
EARSNET_CROP_ENABLED = True

# Neural network model settings
EARSNET_ENABLED = True

# Processing settings
BATCH_SIZE = 16
NUM_WORKERS = 4
CACHE_DIR = "./cache"

# Video configuration
VIDEO_PATH = "./video/Test1.mp4"