Better mobilenetv3 + retinanet model.

datumbox · datumbox · commit 7af35c3874ef · 2021-01-11T19:29:37.000Z
diff --git a/torchvision/models/detection/retinanet.py b/torchvision/models/detection/retinanet.py
@@ -560,7 +560,7 @@ def forward(self, images, targets=None):
 # TODO: replace with pytorch links
 model_urls = {
     'retinanet_mobilenet_v3_large_fpn_coco':
-        'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-575f7893.pth',
+        'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-d88336d5.pth',
     'retinanet_resnet50_fpn_coco':
         'https://download.pytorch.org/models/retinanet_resnet50_fpn_coco-eeacb38b.pth',
 }

Original file line number	Diff line number	Diff line change
`@@ -560,7 +560,7 @@ def forward(self, images, targets=None):`
`560`	`560`	`# TODO: replace with pytorch links`
`561`	`561`	`model_urls = {`
`562`	`562`	`'retinanet_mobilenet_v3_large_fpn_coco':`
`563`		`- 'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-575f7893.pth',`
	`563`	`+ 'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-d88336d5.pth',`
`564`	`564`	`'retinanet_resnet50_fpn_coco':`
`565`	`565`	`'https://download.pytorch.org/models/retinanet_resnet50_fpn_coco-eeacb38b.pth',`
`566`	`566`	`}`