库名称: segmentation-models-pytorch 许可证: mit 管道标签: 图像分割 标签:

• 模型中心混合
• pytorch模型中心混合
• 分割模型-pytorch
• 语义分割
• pytorch 语言:
• python

DeepLabV3Plus 模型卡片

目录:

• 加载训练好的模型
• 模型初始化参数
• 模型指标
• 数据集

加载训练好的模型

import segmentation_models_pytorch as smp

model = smp.from_pretrained("<保存目录或本仓库>")

模型初始化参数

model_init_params = {
    "编码器名称": "efficientnet-b5",
    "编码器深度": 5,
    "编码器权重": "imagenet",
    "编码器输出步长": 16,
    "解码器通道数": 256,
    "解码器空洞率": (12, 24, 36),
    "解码器可分离aspp": True,
    "解码器aspp丢弃率": 0.5,
    "输入通道数": 3,
    "类别数": 1,
    "激活函数": None,
    "上采样倍数": 4,
    "辅助参数": None
}

模型指标

[
    {
        "单图测试iou": 0.866601288318634,
        "数据集测试iou": 0.9076802730560303
    }
]

数据集

数据集名称: 牛津宠物数据集

更多信息

• 库地址: https://github.com/qubvel/segmentation_models.pytorch
• 文档: https://smp.readthedocs.io/en/latest/

本模型通过PytorchModelHubMixin推送至Hub