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 模型，採用來自 Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks 論文的 ResNet-50-FPN 主幹網路。
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 主幹網路。