torchvision.models.detection.faster_rcnn — Torchvision 0.8 ...
pytorch.org/vision/0.8/_modules/torchvision/models/detection/faster_rcnn.htmldef fasterrcnn_resnet50_fpn (pretrained = False, progress = True, num_classes = 91, pretrained_backbone = True, trainable_backbone_layers = 3, ** kwargs): """ Constructs a Faster R-CNN model with a ResNet-50-FPN backbone. The input to the model is expected to be a list of tensors, each of shape ``[C, H, W]``, one for each image, and should be in ``0-1`` range. Different …
PyTorch documentation — PyTorch 1.10.1 documentation
https://pytorch.org/docsPyTorch documentation. PyTorch is an optimized tensor library for deep learning using GPUs and CPUs. Features described in this documentation are classified by release status: Stable: These features will be maintained long-term and there should generally be no major performance limitations or gaps in documentation.