Initial commit: Wood knot detection model and GUI

config.py

@@ -0,0 +1,28 @@
+# 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"
+]