YOLOV4 Pytorch is a PyTorch implementation of the YOLOv4 object detection model for training and running custom detection systems. The repository is structured around practical workflows, including training, prediction, evaluation, anchor generation, model configuration, and dataset annotation utilities. It supports VOC-style datasets and includes scripts for prediction, mAP evaluation, FPS testing, video prediction, batch prediction, and heatmap generation. The project added multi-GPU training, seed settings for reproducible results, adaptive learning rate behavior based on batch size, and both step and cosine learning rate schedules. It also supports Adam and SGD optimizer choices, image cropping, adjustable parameters, and extensive code comments. It is a useful educational and applied repository for users who want to understand or customize YOLOv4 in PyTorch.
Features
- YOLOv4 object detection implementation
- Custom dataset training support
- VOC annotation workflow
- Multi-GPU training support
- mAP and FPS evaluation tools
- Video and batch prediction support