Add FasterRCNN improved weights#5763
Conversation
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
|
@datumbox Could you please provide the link to the |
|
@santhoshnumberone Yes but keep in mind that the FCOS was not retrained. It's likely we can do better with it as well if we review the training recipe but that will be reviewed on the future.
The description of this PR gives you all this info. Use the latest main of TorchVision. |
|
@datumbox off the topic Can't wait to see this DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection paper retrained by you in future |
|
@santhoshnumberone We have a dedicated issue for getting such requests at #2707 I strongly recommend commenting the same thing there so that you get visibility and consider it on the future. Having this discussion here is likely to be forgotten/lost. |
|
Hi @datumbox, I'm trying to reproduce this result (fasterrcnn_resnet50_fpn_v2, mAP=46.7) with torchvision 0.19.1. I have two questions:
Thank you for your help and clarification. |

Fixes #5307
Adds new pre-trained weights for FasterRCNN + ResNet50 + FPN for the v2 variant with post-paper optimizations (no FrozenBN + c5 instead of p5 input on extra layers + heavier RPN/Box Heads with BNs). It improves the previous baseline by +9.7 mAP.
Trained with:
Verified with: