29 lines
746 B
Python
29 lines
746 B
Python
# 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"
|
|
]
|