Mask R-CNN

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

警告

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

模型構建器

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

maskrcnn_resnet50_fpn(*[, weights, ...])	使用 Mask R-CNN 論文中 ResNet-50-FPN 主幹的 Mask R-CNN 模型。
maskrcnn_resnet50_fpn_v2(*[, weights, ...])	使用 Benchmarking Detection Transfer Learning with Vision Transformers 論文中 ResNet-50-FPN 主幹的改進版 Mask R-CNN 模型。