junxnone / aiwiki

AI Wiki

https://junxnone.github.io/aiwiki

17 stars 2 forks source link

ML Tasks Image Detection YOLO V5 #379

Open junxnone opened 1 year ago

junxnone commented 1 year ago

YOLO V5

Reference

YOLOv5 (6.0/6.1) brief summary
Github Code
Docs Wiki
Ultralytics 提供了个数据集管理模型管理的工具 - hub [Trained Model Preview]

Arch

Backbone: New CSP-Darknet53
Neck: SPPF, New CSP-PAN
Head: YOLOv3 Head

Preprocess

Augmentation

Mosaic
Copy-Paste
Mixup
Random horizontal flip
...

Training

Multi-scale training(0.5~1.5x)
AutoAnchor(For training custom data)
Warmup and Cosine LR scheduler
EMA(Exponential Moving Average)
Mixed precision
Evolve hyper-parameters