# config.py - Default settings for annotation GUI

# Default directories (edit these to your preferences)
DEFAULT_IMAGES_DIR = "IMAGE/"  # Directory containing wood defect images
DEFAULT_MODEL_WEIGHTS = "runs/yolox_training/training/weights/best.pt"  # Trained YOLOX model

# Training defaults
DEFAULT_TRAIN_EPOCHS = 20
DEFAULT_BATCH_SIZE = 4
DEFAULT_LEARNING_RATE = 1e-4
DEFAULT_MODEL_SIZE = "small"  # nano, small, medium, base

# Dataset split ratios
DEFAULT_TRAIN_SPLIT = 0.8
DEFAULT_VALID_SPLIT = 0.1
# test split = 1 - train - valid

# GUI settings
DEFAULT_PORT = 7860
DEFAULT_DETECTION_THRESHOLD = 0.5

# Annotation categories (add more if needed)
ANNOTATION_CATEGORIES = [
    "knot",
    # "crack",
    # "hole",
    # "discoloration"
]