Shortcuts

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources