yhenon/pytorch-retinanet
PyTorch implementation of RetinaNet object detection model with ResNet backbone.

Velocity · 7d
+0.7
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of RetinaNet as described in the Focal Loss paper by Lin et al. It is designed for dense object detection tasks and supports training on COCO format datasets or custom CSV-annotated datasets. The model achieves competitive mAP scores (33.5% at 600px with ResNet-50) and includes pre-trained weights for download.