Unverified Commit ddb342dd authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Fix SSD documentation (#6002)

parent e722e9c7
SSD300
======
SSD
===
.. currentmodule:: torchvision.models.detection
The SSD300 model is based on the `SSD: Single Shot MultiBox Detector
The SSD model is based on the `SSD: Single Shot MultiBox Detector
<https://arxiv.org/abs/1512.02325>`__ paper.
Model builders
--------------
The following model builders can be used to instanciate a SSD300 model, with or
The following model builders can be used to instanciate a SSD model, with or
without pre-trained weights. All the model builders internally rely on the
``torchvision.models.detection.SSD`` base class. Please refer to the `source
code
......
......@@ -51,7 +51,6 @@ weights:
models/resnext
models/shufflenetv2
models/squeezenet
models/ssd300
models/swin_transformer
models/vgg
models/vision_transformer
......@@ -127,6 +126,7 @@ weights:
models/faster_rcnn
models/fcos
models/retinanet
models/ssd
models/ssdlite
Table of all available Object detection weights
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment