smp-hub
/

resnet18.ssl

Image Classification

segmentation-models-pytorch

Model card Files Files and versions Community

Model card for resnet18.

This repository contains the ssl 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="ssl")

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_supervised_resnet18-d92f0530.pth

Downloads last month: 17

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.