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

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

Unet模型卡片

目录:

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

加载训练好的模型

import segmentation_models_pytorch as smp  

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

模型初始化参数

model_init_params = {  
    "encoder名称": "resnet152",  
    "编码器深度": 5,  
    "编码器权重": "imagenet",  
    "解码器使用批归一化": True,  
    "解码器通道数": [256, 128, 64, 32, 16],  
    "解码器注意力类型": None,  
    "解码器插值模式": "nearest",  
    "输入通道数": 3,  
    "类别数": 1,  
    "激活函数": None,  
    "辅助参数": None  
}  

模型指标

[  
    {  
        "单张测试图像交并比": 0.9223644733428955,  
        "测试数据集交并比": 0.9241371154785156  
    }  
]  

数据集

数据集名称: IPD

更多信息

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

本模型通过PytorchModelHubMixin推送至Hub