efficientnet-b4.advprop

82 downloads 0 likes on ModelScope updated

Model card


library_name: segmentation-models-pytorch license: other pipeline_tag: image-classification tags:

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

Model card for efficientnet-b4.

This repository contains the advprop pre-trained weights for the efficientnet-b4 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-b4", encoder_weights="advprop")

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/adv-efficientnet-b4-44fb3a87.pth