Skip to content

Latest commit

 

History

History
27 lines (20 loc) · 880 Bytes

ssdlite.rst

File metadata and controls

27 lines (20 loc) · 880 Bytes
Raw

SSDlite

.. currentmodule:: torchvision.models.detection

The SSDLite model is based on the SSD: Single Shot MultiBox Detector, Searching for MobileNetV3 and MobileNetV2: Inverted Residuals and Linear Bottlenecks <https://arxiv.org/abs/1801.04381>__ papers.

.. betastatus:: detection module

Model builders

The following model builders can be used to instantiate a SSD Lite model, with or without pre-trained weights. All the model builders internally rely on the torchvision.models.detection.ssd.SSD base class. Please refer to the source code for more details about this class.

.. autosummary::
    :toctree: generated/
    :template: function.rst

    ssdlite320_mobilenet_v3_large