From 4c5a36cb6a20735cb6c8502fe0556124e5c13e4a Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 6 Apr 2022 13:03:15 +0100 Subject: [PATCH 1/3] Add FasterRCNN improved weights --- torchvision/models/detection/faster_rcnn.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/torchvision/models/detection/faster_rcnn.py b/torchvision/models/detection/faster_rcnn.py index 91aacba7a0a..091094890f5 100644 --- a/torchvision/models/detection/faster_rcnn.py +++ b/torchvision/models/detection/faster_rcnn.py @@ -393,7 +393,17 @@ 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, + "num_params": 43712278, + "recipe": "", + "map": 46.7, + }, + ) + DEFAULT = COCO_V1 class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum): From 54e55901445b43fb285c047aa4f3ddd10c355c96 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 6 Apr 2022 13:06:15 +0100 Subject: [PATCH 2/3] Add recipe URL --- torchvision/models/detection/faster_rcnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/detection/faster_rcnn.py b/torchvision/models/detection/faster_rcnn.py index 091094890f5..53ccf376c2d 100644 --- a/torchvision/models/detection/faster_rcnn.py +++ b/torchvision/models/detection/faster_rcnn.py @@ -399,7 +399,7 @@ class FasterRCNN_ResNet50_FPN_V2_Weights(WeightsEnum): meta={ **_COMMON_META, "num_params": 43712278, - "recipe": "", + "recipe": "https://github.com/pytorch/vision/pull/5763", "map": 46.7, }, ) From 41123846a849a5a4a59f378eddf3af6ba3928687 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 6 Apr 2022 14:37:58 +0100 Subject: [PATCH 3/3] Update publication_year field --- torchvision/models/detection/faster_rcnn.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchvision/models/detection/faster_rcnn.py b/torchvision/models/detection/faster_rcnn.py index 53ccf376c2d..4334d3e4f67 100644 --- a/torchvision/models/detection/faster_rcnn.py +++ b/torchvision/models/detection/faster_rcnn.py @@ -398,6 +398,7 @@ class FasterRCNN_ResNet50_FPN_V2_Weights(WeightsEnum): transforms=ObjectDetection, meta={ **_COMMON_META, + "publication_year": 2021, "num_params": 43712278, "recipe": "https://github.com/pytorch/vision/pull/5763", "map": 46.7,