← all repositories

yhenon/pytorch-retinanet

PyTorch implementation of RetinaNet object detection model with ResNet backbone.

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

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