diff --git a/docs/source/models/ssd300.rst b/docs/source/models/ssd.rst similarity index 75% rename from docs/source/models/ssd300.rst rename to docs/source/models/ssd.rst index a9ceadcdf06..e240d34b178 100644 --- a/docs/source/models/ssd300.rst +++ b/docs/source/models/ssd.rst @@ -1,16 +1,16 @@ -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 `__ 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 diff --git a/docs/source/models_new.rst b/docs/source/models_new.rst index 31b349ac37b..db511425cb4 100644 --- a/docs/source/models_new.rst +++ b/docs/source/models_new.rst @@ -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