bubbliiiing/yolov5-pytorch
PyTorch implementation of YOLOv5 object detection for training custom models.

Velocity · 7d
+0.7
★ / day
Trend
→steady
star history
A complete PyTorch implementation of YOLOv5 (You Only Look Once) object detection algorithm by Ultralytics. The repository supports training custom object detection models with features including multi-GPU training, multiple model sizes (s/m/l/x), EMA, cosine annealing and step learning rate schedulers, Adam/SGD optimizers, and learning rate auto-adjustment based on batch size.