Update stats and model url.

pytorch · Jan 13, 2021 · 110b016 · 110b016
1 parent 54c46ef
commit 110b016
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/source/models.rst b/docs/source/models.rst
@@ -363,7 +363,7 @@ Network                           box AP   mask AP   keypoint AP
 ================================  =======  ========  ===========
 Faster R-CNN ResNet-50 FPN        37.0     -         -
 RetinaNet ResNet-50 FPN           36.4     -         -
-RetinaNet MobileNetV3-Large       ????     -         -
+RetinaNet MobileNetV3-Large       22.1     -         -
 Mask R-CNN ResNet-50 FPN          37.9     34.6      -
 ================================  =======  ========  ===========
 
@@ -420,7 +420,7 @@ Network                         train time (s / it)  test time (s / it)  memory
 ==============================  ===================  ==================  ===========
 Faster R-CNN ResNet-50 FPN      0.2288               0.0590              5.2
 RetinaNet ResNet-50 FPN         0.2514               0.0939              4.1
-RetinaNet MobileNetV3-Large     ??????               ??????              ???
+RetinaNet MobileNetV3-Large     0.0873               0.0408              0.9
 Mask R-CNN ResNet-50 FPN        0.2728               0.0903              5.4
 Keypoint R-CNN ResNet-50 FPN    0.3789               0.1242              6.8
 ==============================  ===================  ==================  ===========

diff --git a/torchvision/models/detection/retinanet.py b/torchvision/models/detection/retinanet.py
@@ -559,7 +559,8 @@ def forward(self, images, targets=None):
 
 # TODO: replace with pytorch links
 model_urls = {
-    'retinanet_mobilenet_v3_large_coco': None,
+    'retinanet_mobilenet_v3_large_coco':
+        'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large-2ae03128.pth',
     'retinanet_resnet50_fpn_coco':
         'https://download.pytorch.org/models/retinanet_resnet50_fpn_coco-eeacb38b.pth',
 }