Keypoint R-CNN¶
Keypoint R-CNN 模型基於 Mask R-CNN 論文。
警告
檢測模組處於 Beta 階段,不保證向後相容。
模型構建器¶
以下模型構建器可用於例項化 Keypoint R-CNN 模型,無論是否包含預訓練權重。所有模型構建器都內部依賴於 torchvision.models.detection.KeypointRCNN 基類。有關此類的更多詳細資訊,請參閱 原始碼。
|
構建一個帶有 ResNet-50-FPN 主幹的 Keypoint R-CNN 模型。 |