库名称：segmentation-models-pytorch
许可证：mit
任务类型：image-segmentation
标签：

• model_hub_mixin
• pytorch_model_hub_mixin
• segmentation-models-pytorch
• semantic-segmentation
• pytorch
  支持语言：
• python

DPT模型卡片

目录：

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

加载训练好的模型

import segmentation_models_pytorch as smp  

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

模型初始化参数

model_init_params = {  
    "encoder_name": "tu-vit_large_patch16_384",  
    "encoder_depth": 4,  
    "encoder_weights": None,  
    "feature_dim": 256,  
    "in_channels": 3,  
    "classes": 150,  
    "activation": None,  
    "aux_params": None,  
    "output_stride": None  
}  

模型指标

[需要更多信息]

数据集

数据集名称：[需要更多信息]

更多信息

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

本模型通过PytorchModelHubMixin推送至Hub