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 模型，配備來自 Mask R-CNN 論文的 ResNet-50-FPN 主幹網路。
maskrcnn_resnet50_fpn_v2(*[, weights, ...])	改進的 Mask R-CNN 模型，配備來自 Benchmarking Detection Transfer Learning with Vision Transformers 論文的 ResNet-50-FPN 主幹網路。