← all repositories

tonylins/pytorch-mobilenet-v2

A PyTorch implementation of the MobileNetV2 convolutional neural network architecture with pretrained model weights.

1.4k stars Python Computer VisionML Frameworks
pytorch-mobilenet-v2
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

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

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.