快捷方式

DeepLabV3

DeepLabV3 模型基于 Rethinking Atrous Convolution for Semantic Image Segmentation 论文。

警告

分割模块处于 Beta 阶段,不保证向后兼容性。

模型构建器

以下模型构建器可用于实例化具有不同骨干网络、带或不带预训练权重的 DeepLabV3 模型。所有模型构建器内部都依赖于 torchvision.models.segmentation.deeplabv3.DeepLabV3 基类。有关此类的更多详细信息,请参阅源代码

deeplabv3_mobilenet_v3_large(*[, weights, ...])

构建一个带有 MobileNetV3-Large 骨干网络的 DeepLabV3 模型。

deeplabv3_resnet50(*[, weights, progress, ...])

构建一个带有 ResNet-50 骨干网络的 DeepLabV3 模型。

deeplabv3_resnet101(*[, weights, progress, ...])

构建一个带有 ResNet-101 骨干网络的 DeepLabV3 模型。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源