diff --git a/config/app_config.yaml b/config/app_config.yaml index 5fbb942..9d15cbf 100644 --- a/config/app_config.yaml +++ b/config/app_config.yaml @@ -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