Implementing 2 stage training

2025-12-11 12:04:08 +02:00
parent 221c80aa8c
commit c0684a9c14
5 changed files with 315 additions and 5 deletions
--- a/config/app_config.yaml
+++ b/config/app_config.yaml
@@ -12,12 +12,26 @@ image_repository:
 models:
  default_base_model: yolov8s-seg.pt
  models_directory: data/models
+  base_model_choices:
+  - yolov8s-seg.pt
+  - yolov11s-seg.pt
 training:
  default_epochs: 100
  default_batch_size: 16
  default_imgsz: 640
  default_patience: 50
  default_lr0: 0.01
+  two_stage:
+    enabled: false
+    stage1:
+      epochs: 20
+      lr0: 0.0005
+      patience: 10
+      freeze: 10
+    stage2:
+      epochs: 150
+      lr0: 0.0003
+      patience: 30
  last_dataset_yaml: /home/martin/code/object_detection/data/datasets/data.yaml
  last_dataset_dir: /home/martin/code/object_detection/data/datasets
 detection: