smp-hub/timm-efficientnet-b5.imagenet

Model card for timm-efficientnet-b5.

This repository contains the imagenet pre-trained weights for the timm-efficientnet-b5 model used as encoder in the segmentation-models-pytorch library.

Example usage:

1. Install the library:

pip install segmentation-models-pytorch

2. Use the encoder in your code:

import segmentation_models_pytorch as smp

model = smp.Unet("timm-efficientnet-b5", encoder_weights="imagenet")

References

• Github: https://github.com/qubvel/segmentation_models.pytorch
• Docs: https://smp.readthedocs.io/en/latest/
• Original weights URL: https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b5-c6949ce9.pth