许可证: mit
库名称: segmentation-models-pytorch
标签:

• 语义分割
• pytorch
• segmentation-models-pytorch
  语言:
• python
  流水线标签: 图像分割

FPN模型卡片

目录:

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

加载训练好的模型

import segmentation_models_pytorch as smp  

model = smp.FPN.from_pretrained("oxford-pet-segmentation")  

模型初始化参数

model_init_params = {  
    "encoder_name": "resnet34",  
    "encoder_depth": 5,  
    "encoder_weights": "imagenet",  
    "decoder_pyramid_channels": 256,  
    "decoder_segmentation_channels": 128,  
    "decoder_merge_policy": "add",  
    "decoder_dropout": 0.2,  
    "in_channels": 3,  
    "classes": 1,  
    "activation": None,  
    "upsampling": 4,  
    "aux_params": None  
}  

模型指标

[  
    {  
        "test_per_image_iou": 0.9065093398094177,  
        "test_dataset_iou": 0.9125494360923767  
    }  
]  

数据集

数据集名称: Oxford Pet

更多信息

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

此模型已通过PytorchModelHubMixin推送至Hub