diff --git a/docs/source/models.rst b/docs/source/models.rst index f27a555befe..c79b89d5f71 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -201,6 +201,12 @@ ResNext Semantic Segmentation ===================== +The models subpackage contains definitions for the following model +architectures for semantic segmentation: + +- `FCN ResNet101 `_ +- `DeepLabV3 ResNet101 `_ + As with image classification models, all pre-trained models expect input images normalized in the same way. The images have to be loaded in to a range of ``[0, 1]`` and then normalized using ``mean = [0.485, 0.456, 0.406]`` and ``std = [0.229, 0.224, 0.225]``. @@ -244,6 +250,12 @@ DeepLabV3 Object Detection, Instance Segmentation and Person Keypoint Detection ===================================================================== +The models subpackage contains definitions for the following model +architectures for detection: + +- `Faster R-CNN ResNet-50 FPN `_ +- `Mask R-CNN ResNet-50 FPN `_ + The pre-trained models for detection, instance segmentation and keypoint detection are initialized with the classification models in torchvision.