smp-hub/resnet152.imagenet
02.8k
libraryname: segmentation-models-pytorch license: other pipelinetag: image-classification tags:
- segmentation-models-pytorch
- image-classification
- pytorch
- resnet languages:
- python ---
Model card for resnet152.
This repository contains the imagenet pre-trained weights for the resnet152 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("resnet152", encoder_weights="imagenet")References
- Github: https://github.com/qubvel/segmentation_models.pytorch
- Docs: https://smp.readthedocs.io/en/latest/
- Original weights URL: https://download.pytorch.org/models/resnet152-b121ed2d.pth
