# 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 = (255, 165, 0)  # Green
XGBOOST_COLOR = (255, 0, 0)  # Red
LIGHTGBM_COLOR = (0, 0, 255)  # Blue
RESNET_COLOR = (255, 165, 0)  # Orange
EFFICIENTNET_COLOR = (0, 0, 255)  # Blue
MOBILENET_COLOR = (255, 0, 0)  # Red
SQUEEZENET_COLOR = (128, 0, 128)  # Purple

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

# Neural network model settings
RESNET_ENABLED = False
EFFICIENTNET_ENABLED = False
MOBILENET_ENABLED = False
SQUEEZENET_ENABLED = False

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

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