# 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 (R,G,B format)
CONV_COLOR = (0, 255, 0)  # Green
XGBOOST_COLOR = (255, 0, 0)  # Red
LIGHTGBM_COLOR = (0, 0, 255)  # Blue
EARSNET_COLOR = (0, 165, 255)  # Orange

# Model execution settings
CONV_ENABLED = True
XGBOOST_ENABLED = False
LIGHTGBM_ENABLED = True
NORMALIZE_ENABLED = False
POSENET_ENABLED = False
RTMPOSE_ENABLED = True
MOBILENETV1SSD_ENABLED = False
YOLOX_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"