metadata

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 library.

Example usage:

Install the library:

pip install segmentation-models-pytorch

Use the encoder in your code:

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