58 lines
1.1 KiB
YAML
58 lines
1.1 KiB
YAML
database:
|
|
path: data/detections.db
|
|
image_repository:
|
|
base_path: ''
|
|
allowed_extensions:
|
|
- .jpg
|
|
- .jpeg
|
|
- .png
|
|
- .tif
|
|
- .tiff
|
|
- .bmp
|
|
models:
|
|
default_base_model: yolov8s-seg.pt
|
|
models_directory: data/models
|
|
base_model_choices:
|
|
- yolov8s-seg.pt
|
|
- yolo11s-seg.pt
|
|
training:
|
|
default_epochs: 100
|
|
default_batch_size: 16
|
|
default_imgsz: 1024
|
|
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:
|
|
default_confidence: 0.25
|
|
default_iou: 0.45
|
|
max_batch_size: 100
|
|
visualization:
|
|
bbox_colors:
|
|
organelle: '#FF6B6B'
|
|
membrane_branch: '#4ECDC4'
|
|
default: '#00FF00'
|
|
bbox_thickness: 2
|
|
font_size: 12
|
|
export:
|
|
formats:
|
|
- csv
|
|
- json
|
|
- excel
|
|
default_format: csv
|
|
logging:
|
|
level: INFO
|
|
file: logs/app.log
|
|
format: '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|