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, ...])	改進的 Mask R-CNN 模型，具有來自 使用 Vision Transformers 進行基準檢測轉移學習 論文的 ResNet-50-FPN 主幹。