Linknet Model Card

Table of Contents:

Load trained model

import segmentation_models_pytorch as smp

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

Model init parameters

model_init_params = {
    "encoder_name": "resnet34",
    "encoder_depth": 5,
    "encoder_weights": "imagenet",
    "decoder_use_batchnorm": True,
    "in_channels": 1,
    "classes": 1,
    "activation": None,
    "aux_params": None
}

Model metrics

[
    {
        "test_per_image_iou": 0.760050892829895,
        "test_dataset_iou": 0.7403926253318787
    }
]

Dataset

Dataset name: Breast

More Information

This model has been pushed to the Hub using the PytorchModelHubMixin

Downloads last month
169
Safetensors
Model size
21.8M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support