Faster R-CNN

Also written: fasterrcnn_resnet50_fpn_v2, FastRCNNPredictor

A pretrained object detector in torchvision. It takes a list of images and returns boxes, labels and scores. To train your own, you swap its box predictor head for one with your number of classes, counting "background" as a class.