快捷方式

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(*[, ...])

構建一個帶有 MobileNetV3-Large FPN 主幹網路的高解析度 Faster R-CNN 模型。

fasterrcnn_mobilenet_v3_large_320_fpn(*[, ...])

一個帶有 MobileNetV3-Large 主幹網路的低解析度 Faster R-CNN 模型,針對移動用例進行了調整。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源