ssdlite.rst 880 B

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