Skip to content

Commit 3f62b5c

Browse files
authored
Add FasterRCNN improved weights (#5763)
* Add FasterRCNN improved weights * Add recipe URL * Update publication_year field
1 parent b5481e4 commit 3f62b5c

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)