捷徑

Faster R-CNN

Faster R-CNN 模型基於 Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks 論文。

警告

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

模型建構器

以下模型建構器可用於實例化 Faster R-CNN 模型,帶有或不帶有預先訓練的權重。所有模型建構器都在內部依賴 torchvision.models.detection.faster_rcnn.FasterRCNN 基底類別。有關此類別的更多詳細資訊,請參閱原始程式碼

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

具有來自 Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks 論文的 ResNet-50-FPN 骨幹網路的 Faster R-CNN 模型。

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

建構一個改進的 Faster R-CNN 模型,其具有來自 Benchmarking Detection Transfer Learning with Vision Transformers 論文的 ResNet-50-FPN 骨幹網路。

fasterrcnn_mobilenet_v3_large_fpn(*[, ...])

建構一個高解析度的 Faster R-CNN 模型,其具有 MobileNetV3-Large FPN 骨幹網路。

fasterrcnn_mobilenet_v3_large_320_fpn(*[, ...])

低解析度的 Faster R-CNN 模型,其具有針對行動用例調整的 MobileNetV3-Large 骨幹網路。

文件

存取 PyTorch 的全面開發者文件

查看文件

教學

取得針對初學者和進階開發者的深入教學

查看教學

資源

尋找開發資源並獲得解答

檢視資源