Updating config
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user