# 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 = (110, 32, 120)  # 純粋な緑
XGBOOST_COLOR = (19, 80, 27)  # オレンジ
LIGHTGBM_COLOR = (19, 80, 27)  # マゼンタ
EARSNET_COLOR = (192, 79, 21)  # ダークブルー
EARSNET_CROP_COLOR = (192, 79, 21)
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 = False
RTMPOSE_ENABLED = True
MOBILENETV1SSD_ENABLED = False
YOLOX_ENABLED = True
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"