roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

RetinaNet support #183

Open miliadis opened 5 years ago

miliadis commented 5 years ago

This is a PR to support RetinaNet, a state-of-the-art one-stage object detector. The model (ResNet 50 with 800 pixels) achieved AP of 35.3 instead of 35.7 reported on the paper.

  1. you can download my PyTorch weights here (AP 35.3)
  2. this code allows you also to load Caffe2's weights of RetinaNet. You can download the weights here (AP 35.7)

Details of results are shown below:

RetinaNet

retinanet-R-50-FPN_1x

Box
source AP50:95 AP50 AP75 APs APm APl AR1 AR10 AR100 ARs ARm ARl
PyTorch 35.3 54.6 37.9 19.4 39.1 47.5 30.7 48.9 51.8 32.4 56.3 67.4
Detectron 35.7 54.7 38.5 19.5 39.9 47.5 30.7 49.1 52.0 32.0 56.9 68.0