bubbliiiing/yolov4-pytorch
A PyTorch implementation of the YOLOv4 object detection model with support for training custom models.

Velocity · 7d
+1.0
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of YOLOv4, a widely-used one-stage object detection architecture. It includes the full training pipeline with support for multi-GPU training, various optimizers (SGD, Adam), learning rate schedulers (step, cosine), and batch-size-adaptive learning rates. The project also supports inference on images, videos, and batch predictions with FPS measurement.