Shortcuts

SSDlite

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.

Warning

The detection module is in Beta stage, and backward compatibility is not guaranteed.

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.

ssdlite320_mobilenet_v3_large(*[, weights, ...])

SSDlite model architecture with input size 320x320 and a MobileNetV3 Large backbone, as described at Searching for MobileNetV3 and MobileNetV2: Inverted Residuals and Linear Bottlenecks.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources