import torchvision from torchvision.models.detection.faster_rcnn import FastRCNNPredictor # load a model pre-trained pre-trained on COCO model = torchvision. models. detection. fasterrcnn_resnet50_fpn (pretrained = True) # replace the classifier with a new one, that has # num_classes which is user-defined num_classes = 2 # 1 class (person ...