Keypoint R-CNN¶
The Keypoint R-CNN model is based on the Mask R-CNN 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 Keypoint R-CNN model,
with or without pre-trained weights. All the model builders internally rely on
the torchvision.models.detection.KeypointRCNN
base class. Please refer to the source
code
for more details about this class.
|
Constructs a Keypoint R-CNN model with a ResNet-50-FPN backbone. |