← all repositories

bubbliiiing/yolov4-pytorch

A PyTorch implementation of the YOLOv4 object detection model with support for training custom models.

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

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