CoolFace
Modelpublic

smp-hub/timm-efficientnet-b4.imagenet

sourceHugging Faceotherupdated 2y agoView on Hugging Face
0likes1.2kdownloads
Model Card

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

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

Model card for timm-efficientnet-b4.

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

Example usage:

  1. 1.Install the library:
bash
pip install segmentation-models-pytorch
  1. 1.Use the encoder in your code:
python
import segmentation_models_pytorch as smp

model = smp.Unet("timm-efficientnet-b4", 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/tfefficientnetb4-74ee3bed.pth