yolov5/data/hyp.finetune.yaml

35 lines
772 B
YAML
Raw Normal View History

# Hyperparameters for VOC finetuning
# python train.py --batch 64 --weights yolov5m.pt --data voc.yaml --img 512 --epochs 50
# See tutorials for hyperparameter evolution https://github.com/ultralytics/yolov5#tutorials
# Hyperparameter Evolution Results
# Generations: 51
# P R mAP.5 mAP.5:.95 box obj cls
# Metrics: 0.625 0.926 0.89 0.677 0.0111 0.00849 0.00124
lr0: 0.00447
lrf: 0.114
momentum: 0.873
weight_decay: 0.00047
giou: 0.0306
cls: 0.211
cls_pw: 0.546
obj: 0.421
obj_pw: 0.972
iou_t: 0.2
anchor_t: 2.26
# anchors: 5.07
fl_gamma: 0.0
hsv_h: 0.0154
hsv_s: 0.9
hsv_v: 0.619
degrees: 0.404
translate: 0.206
scale: 0.86
shear: 0.795
perspective: 0.0
flipud: 0.00756
fliplr: 0.5
mixup: 0.153