快捷方式

DeepLabV3

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

警告

The segmentation module is in Beta stage, and backward compatibility is not guaranteed.

模型构建器

以下模型构建器可用于实例化具有不同骨干网络、包含或不包含预训练权重的 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 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源