Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add FasterRCNN improved weights #5763

Merged
merged 4 commits into from
Apr 6, 2022
Merged

Add FasterRCNN improved weights #5763

Changes from 3 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
4c5a36c
Add FasterRCNN improved weights
datumbox Apr 6, 2022
54e5590
Add recipe URL
datumbox Apr 6, 2022
4112384
Update publication_year field
datumbox Apr 6, 2022
468dc32
Merge branch 'main' into modes/fasterrcnn
datumbox Apr 6, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 12 additions & 1 deletion torchvision/models/detection/faster_rcnn.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down