Faster R-CNN

The Faster R-CNN model is based on the Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks paper.

Warning

The detection module is in Beta stage, and backward compatibility is not guaranteed.

Model builders

The following model builders can be used to instantiate a Faster R-CNN model, with or without pre-trained weights. All the model builders internally rely on the torchvision.models.detection.faster_rcnn.FasterRCNN base class. Please refer to the source code for more details about this class.

fasterrcnn_resnet50_fpn(*[, weights, …])	Faster R-CNN model with a ResNet-50-FPN backbone from the Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks paper.
fasterrcnn_resnet50_fpn_v2(*[, weights, …])	Constructs an improved Faster R-CNN model with a ResNet-50-FPN backbone from Benchmarking Detection Transfer Learning with Vision Transformers paper.
fasterrcnn_mobilenet_v3_large_fpn(*[, …])	Constructs a high resolution Faster R-CNN model with a MobileNetV3-Large FPN backbone.
fasterrcnn_mobilenet_v3_large_320_fpn(*[, …])	Low resolution Faster R-CNN model with a MobileNetV3-Large backbone tunned for mobile use cases.