smp-hub
/

resnet18.swsl

Image Classification

segmentation-models-pytorch

Model card Files Files and versions Community

Model card for resnet18.

This repository contains the swsl pre-trained weights for the resnet18 model used as encoder in the segmentation-models-pytorch library.

Example usage:

Install the library:

pip install segmentation-models-pytorch

Use the encoder in your code:

import segmentation_models_pytorch as smp

model = smp.Unet("resnet18", encoder_weights="swsl")

References

Github: https://github.com/qubvel/segmentation_models.pytorch
Docs: https://smp.readthedocs.io/en/latest/
Original weights URL: https://dl.fbaipublicfiles.com/semiweaksupervision/model_files/semi_weakly_supervised_resnet18-118f1556.pth

Downloads last month: 16

Safetensors

Model size

11.7M params

Tensor type

F32

·

Inference Providers NEW

Image Classification

This model is not currently available via any of the supported Inference Providers.

The model cannot be deployed to the HF Inference API: The HF Inference API does not support image-classification models for segmentation-models-pytorch library.