Add FasterRCNN improved weights (#5763)

* Add FasterRCNN improved weights * Add recipe URL * Update publication_year field

Add FasterRCNN improved weights (#5763)
* Add FasterRCNN improved weights * Add recipe URL * Update publication_year field
3f62b5c0 · Vasilis Vryniotis · GitHub · b5481e49 · 3f62b5c0
Unverified Commit 3f62b5c0 authored Apr 06, 2022 by Vasilis Vryniotis Committed by GitHub Apr 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 1 deletion

torchvision/models/detection/faster_rcnn.py torchvision/models/detection/faster_rcnn.py +12 -1

No files found.
--- a/torchvision/models/detection/faster_rcnn.py
+++ b/torchvision/models/detection/faster_rcnn.py
@@ -393,7 +393,18 @@ class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum):


 class FasterRCNN_ResNet50_FPN_V2_Weights(WeightsEnum):
-    pass
+    COCO_V1 = Weights(
+        url="https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_v2_coco-dd69338a.pth",
+        transforms=ObjectDetection,
+        meta={
+            **_COMMON_META,
+            "publication_year": 2021,
+            "num_params": 43712278,
+            "recipe": "https://github.com/pytorch/vision/pull/5763",
+            "map": 46.7,
+        },
+    )
+    DEFAULT = COCO_V1


 class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum):