/MobileNetV2-pytorch

Impementation of MobileNetV2 in pytorch

Primary LanguagePythonMIT LicenseMIT

MobileNet2 in PyTorch

An implementation of MobileNet2 in PyTorch. MobileNet2 is an efficient convolutional neural network architecture for mobile devices. For more information check the paper: Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation

Usage

Clone the repo:

git clone https://github.com/Randl/MobileNet2-pytorch/
pip install -r requirements.txt

Use the model defined in model.py to run ImageNet example:

python imagenet.py --dataroot "/path/to/imagenet/"