← all repositories

bubbliiiing/yolov5-pytorch

PyTorch implementation of YOLOv5 object detection for training custom models.

1.1k stars Python Computer VisionML Frameworks
yolov5-pytorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.