smp-hub/efficientnet-b0.imagenet

image classificationsegmentation-models-pytorchsegmentation-models-pytorchsafetensorsimage-classificationpytorchefficientnetlicense:otherother
171.9K

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. Install the library:
pip install segmentation-models-pytorch
  1. Use the encoder in your code:
import segmentation_models_pytorch as smp

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

References

DEPLOY IN 60 SECONDS

Run efficientnet-b0.imagenet on Runcrate

Deploy on H100, A100, or RTX GPUs. Pay only for what you use. No setup required.