ssd.rst 678 B

1234567891011121314151617181920212223242526
  1. SSD
  2. ===
  3. .. currentmodule:: torchvision.models.detection
  4. The SSD model is based on the `SSD: Single Shot MultiBox Detector
  5. <https://arxiv.org/abs/1512.02325>`__ paper.
  6. .. betastatus:: detection module
  7. Model builders
  8. --------------
  9. The following model builders can be used to instantiate a SSD model, with or
  10. without pre-trained weights. All the model builders internally rely on the
  11. ``torchvision.models.detection.SSD`` base class. Please refer to the `source
  12. code
  13. <https://github.com/pytorch/vision/blob/main/torchvision/models/detection/ssd.py>`_ for
  14. more details about this class.
  15. .. autosummary::
  16. :toctree: generated/
  17. :template: function.rst
  18. ssd300_vgg16