Keypoint R-CNN

Keypoint R-CNN 模型基於 Mask R-CNN 論文。

警告

檢測模組處於 Beta 階段，不保證向後相容性。

模型構建器

以下模型構建器可用於例項化 Keypoint R-CNN 模型，可選擇是否使用預訓練權重。所有模型構建器內部都依賴於 torchvision.models.detection.KeypointRCNN 基類。有關此類的更多詳細資訊，請參閱原始碼。

keypointrcnn_resnet50_fpn(*[, weights, ...])

使用 ResNet-50-FPN 主幹構建 Keypoint R-CNN 模型。