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

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.