| library_name: segmentation-models-pytorch | |
| license: other | |
| pipeline_tag: image-classification | |
| tags: | |
| - segmentation-models-pytorch | |
| - image-classification | |
| - pytorch | |
| - efficientnet | |
| languages: | |
| - python | |
| # Model card for timm-efficientnet-b0. | |
| This repository contains the `advprop` pre-trained weights for the `timm-efficientnet-b0` model used as | |
| encoder in the [segmentation-models-pytorch](https://github.com/qubvel-org/segmentation_models.pytorch) library. | |
| ### Example usage: | |
| 1. Install the library: | |
| ```bash | |
| pip install segmentation-models-pytorch | |
| ``` | |
| 2. Use the encoder in your code: | |
| ```python | |
| import segmentation_models_pytorch as smp | |
| model = smp.Unet("timm-efficientnet-b0", 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/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b0_ap-f262efe1.pth | |