Dpt Tu Test Vit

🚀 DPT模型卡片

DPT模型主要用于图像分割任务，借助segmentation - models - pytorch库，可方便地加载预训练模型并进行参数初始化，为图像分割工作提供了高效的解决方案。

🚀 快速开始

加载预训练模型

import segmentation_models_pytorch as smp

model = smp.from_pretrained("<save-directory-or-this-repo>")

模型初始化参数

model_init_params = {
    "encoder_name": "tu-test_vit",
    "encoder_depth": 4,
    "encoder_weights": "imagenet",
    "encoder_output_indices": None,
    "decoder_readout": "cat",
    "decoder_intermediate_channels": (16, 32, 64, 64),
    "decoder_fusion_channels": 16,
    "in_channels": 3,
    "classes": 1,
    "activation": None,
    "aux_params": None,
    "dynamic_img_size": True
}

📚 详细文档

更多信息

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

本模型已使用 PytorchModelHubMixin 推送到模型中心。

📄 许可证

本项目采用MIT许可证。