tonylins/pytorch-mobilenet-v2
A PyTorch implementation of the MobileNetV2 convolutional neural network architecture with pretrained model weights.

This repository provides a PyTorch implementation of MobileNetV2, a efficient deep learning architecture designed for mobile and embedded vision tasks. It includes model code replicating the original architecture described in the paper, pretrained weights downloadable automatically, and a training recipe achieving over 72% top-1 accuracy on ImageNet. The implementation matches the official TensorFlow version in FLOPs (~300M) and parameters (3.47M).