faster_rcnn.rst 913 B

12345678910111213141516171819202122232425262728293031
  1. Faster R-CNN
  2. ============
  3. .. currentmodule:: torchvision.models.detection
  4. The Faster R-CNN model is based on the `Faster R-CNN: Towards Real-Time Object Detection
  5. with Region Proposal Networks <https://arxiv.org/abs/1506.01497>`__
  6. paper.
  7. .. betastatus:: detection module
  8. Model builders
  9. --------------
  10. The following model builders can be used to instantiate a Faster R-CNN model, with or
  11. without pre-trained weights. All the model builders internally rely on the
  12. ``torchvision.models.detection.faster_rcnn.FasterRCNN`` base class. Please refer to the `source
  13. code
  14. <https://github.com/pytorch/vision/blob/main/torchvision/models/detection/faster_rcnn.py>`_ for
  15. more details about this class.
  16. .. autosummary::
  17. :toctree: generated/
  18. :template: function.rst
  19. fasterrcnn_resnet50_fpn
  20. fasterrcnn_resnet50_fpn_v2
  21. fasterrcnn_mobilenet_v3_large_fpn
  22. fasterrcnn_mobilenet_v3_large_320_fpn