Skip to content

Commit 345b597

Browse files
YosuaMichaelfacebook-github-bot
authored andcommitted
[fbsync] Add FasterRCNN improved weights (#5763)
Summary: * Add FasterRCNN improved weights * Add recipe URL * Update publication_year field (Note: this ignores all push blocking failures!) Reviewed By: jdsgomes, NicolasHug Differential Revision: D36095675 fbshipit-source-id: e8513c22da8e14419de0749b8085ee587ff11d84
1 parent ffadf7e commit 345b597

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

torchvision/models/detection/faster_rcnn.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,18 @@ class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum):
393393

394394

395395
class FasterRCNN_ResNet50_FPN_V2_Weights(WeightsEnum):
396-
pass
396+
COCO_V1 = Weights(
397+
url="https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_v2_coco-dd69338a.pth",
398+
transforms=ObjectDetection,
399+
meta={
400+
**_COMMON_META,
401+
"publication_year": 2021,
402+
"num_params": 43712278,
403+
"recipe": "https://github.com/pytorch/vision/pull/5763",
404+
"map": 46.7,
405+
},
406+
)
407+
DEFAULT = COCO_V1
397408

398409

399410
class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum):

0 commit comments

Comments
 (0)