Table of Contents
Shortcuts

# fasterrcnn_resnet50_fpn_v2

torchvision.models.detection.fasterrcnn_resnet50_fpn_v2(*, weights: Optional[FasterRCNN_ResNet50_FPN_V2_Weights] = None, progress: bool = True, num_classes: Optional[int] = None, weights_backbone: Optional[ResNet50_Weights] = None, trainable_backbone_layers: Optional[int] = None, **kwargs: Any) → FasterRCNN

  Constructs an improved Faster R-CNN model with a ResNet-50-FPN backbone from Benchmarking Detection Transfer Learning with Vision Transformers paper.

  Warning

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

  It works similarly to Faster R-CNN with ResNet-50 FPN backbone. See `fasterrcnn_resnet50_fpn()` for more details.

  Parameters:

    - weights (`FasterRCNN_ResNet50_FPN_V2_Weights`, optional) – The pretrained weights to use. See `FasterRCNN_ResNet50_FPN_V2_Weights` below for more details, and possible values. By default, no pre-trained weights are used.

    - progress (bool, optional) – If True, displays a progress bar of the download to stderr. Default is True.

    - num_classes (int, optional) – number of output classes of the model (including the background)

    - weights_backbone (`ResNet50_Weights`, optional) – The pretrained weights for the backbone.

    - trainable_backbone_layers (int, optional) – number of trainable (not frozen) layers starting from final block. Valid values are between 0 and 5, with 5 meaning all backbone layers are trainable. If `None` is passed (the default) this value is set to 3.

    - **kwargs – parameters passed to the `torchvision.models.detection.faster_rcnn.FasterRCNN` base class. Please refer to the source code for more details about this class.

  class torchvision.models.detection.FasterRCNN_ResNet50_FPN_V2_Weights(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

## 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

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.