Updating config

This commit is contained in:
2025-12-12 21:51:12 +02:00
parent 9f7d2be1ac
commit 20a87c9040

View File

@@ -14,11 +14,11 @@ models:
models_directory: data/models
base_model_choices:
- yolov8s-seg.pt
- yolov11s-seg.pt
- yolo11s-seg.pt
training:
default_epochs: 100
default_batch_size: 16
default_imgsz: 640
default_imgsz: 1024
default_patience: 50
default_lr0: 0.01
two_stage:
@@ -32,8 +32,8 @@ training:
epochs: 150
lr0: 0.0003
patience: 30
last_dataset_yaml: /home/martin/code/object_detection/data/datasets-revert/data.yaml
last_dataset_dir: /home/martin/code/object_detection/data/datasets-revert
last_dataset_yaml: /home/martin/code/object_detection/data/datasets/data.yaml
last_dataset_dir: /home/martin/code/object_detection/data/datasets
detection:
default_confidence: 0.25
default_iou: 0.45