| library_name: segmentation-models-pytorch | |
| license: other | |
| pipeline_tag: image-classification | |
| tags: | |
| - segmentation-models-pytorch | |
| - image-classification | |
| - pytorch | |
| - mobilenet | |
| languages: | |
| - python | |
| # Model card for mobilenet_v2. | |
| This repository contains the `imagenet` pre-trained weights for the `mobilenet_v2` 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("mobilenet_v2", encoder_weights="imagenet") | |
| ``` | |
| ### References | |
| - Github: https://github.com/qubvel/segmentation_models.pytorch | |
| - Docs: https://smp.readthedocs.io/en/latest/ | |
| - Original weights URL: https://download.pytorch.org/models/mobilenet_v2-b0353104.pth | |