Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "bbf733ab704842401deec1e60be642e4f07404dd"
Unverified Commit 3f62b5c0 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Add FasterRCNN improved weights (#5763) 

* Add FasterRCNN improved weights

* Add recipe URL

* Update publication_year field
parent b5481e49
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 1 deletion
torchvision/models/detection/faster_rcnn.py
View file @ 3f62b5c0
...@@ -393,7 +393,18 @@ class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum): ...@@ -393,7 +393,18 @@ class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum):
class FasterRCNN_ResNet50_FPN_V2_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): class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment