smp-hub/efficientnet-b0.imagenet

libraryname: segmentation-models-pytorch license: other pipelinetag: image-classification tags:

• —segmentation-models-pytorch
• —image-classification
• —pytorch
• —efficientnet languages:
• —python ---

Model card for efficientnet-b0.

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

Example usage:

1. 1.Install the library:

pip install segmentation-models-pytorch

1. 1.Use the encoder in your code:

import segmentation_models_pytorch as smp

model = smp.Unet("efficientnet-b0", 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/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b0-355c32eb.pth