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 .. 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. <https://arxiv.org/abs/1512.02325>`__ paper.
Model builders 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 without pre-trained weights. All the model builders internally rely on the
``torchvision.models.detection.SSD`` base class. Please refer to the `source ``torchvision.models.detection.SSD`` base class. Please refer to the `source
code code
......
...@@ -51,7 +51,6 @@ weights: ...@@ -51,7 +51,6 @@ weights:
models/resnext models/resnext
models/shufflenetv2 models/shufflenetv2
models/squeezenet models/squeezenet
models/ssd300
models/swin_transformer models/swin_transformer
models/vgg models/vgg
models/vision_transformer models/vision_transformer
...@@ -127,6 +126,7 @@ weights: ...@@ -127,6 +126,7 @@ weights:
models/faster_rcnn models/faster_rcnn
models/fcos models/fcos
models/retinanet models/retinanet
models/ssd
models/ssdlite models/ssdlite
Table of all available Object detection weights 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