← all repositories

argusswift/YOLOv4-pytorch

PyTorch re-implementation of YOLOv4 for object detection with attention modules and MobileNet backbone variants.

1.7k stars Python Computer VisionML Frameworks
YOLOv4-pytorch
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of YOLOv4, a popular real-time object detection architecture. It includes attentive variants using SEnet and CBAM attention mechanisms, as well as MobileNetV2 and MobileNetV3 backbone options for faster inference. The project supports training on PASCAL VOC and COCO datasets, includes data augmentation, multi-scale training, and mixed precision (fp16) training capabilities.

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