We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ffadf7e commit 345b597Copy full SHA for 345b597
torchvision/models/detection/faster_rcnn.py
@@ -393,7 +393,18 @@ class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum):
393
394
395
class FasterRCNN_ResNet50_FPN_V2_Weights(WeightsEnum):
396
- pass
+ 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
408
409
410
class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum):
0 commit comments