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