torchvision.models — Torchvision 0.11.0 documentation
pytorch.org › vision › stableSemantic Segmentation¶ The models subpackage contains definitions for the following model architectures for semantic segmentation: FCN ResNet50, ResNet101. DeepLabV3 ResNet50, ResNet101, MobileNetV3-Large. LR-ASPP MobileNetV3-Large. As with image classification models, all pre-trained models expect input images normalized in the same way.