Skip to content

This repository contains the official implementation of the research paper, "An Improved One millisecond Mobile Backbone".

License

Notifications You must be signed in to change notification settings

huyiming2018/ml-mobileone

 
 

Repository files navigation

MobileOne: An Improved One millisecond Mobile Backbone

This software project accompanies the research paper, An Improved One millisecond Mobile Backbone.

Our model achieves Top-1 Accuracy of 75.9% under 1ms.

MobileOne Performance

Model Zoo

ImageNet-1K

Model Top-1 Acc. Latency* Pytorch Checkpoint (url) CoreML Model
MobileOne-S0 71.4 0.79 S0(unfused) mlmodel
MobileOne-S1 75.9 0.89 S1(unfused) mlmodel
MobileOne-S2 77.4 1.18 S2(unfused) mlmodel
MobileOne-S3 78.1 1.53 S3(unfused) mlmodel
MobileOne-S4 79.4 1.86 S4(unfused) mlmodel

*Latency measured on iPhone 12 Pro.

Usage

To use our model, follow the code snippet below,

import torch
from mobileone import mobileone, reparameterize_model

# To Train from scratch/fine-tuning
model = mobileone(variant='s0')
# ... train ...

# Load Pre-trained checkpoint for fine-tuning
checkpoint = torch.load('/path/to/unfused_checkpoint.pth.tar')
model.load_state_dict(checkpoint)
# ... train ...

# For inference
model.eval()      
model_eval = reparameterize_model(model)
# Use model_eval at test-time

To simply evaluate our model, use the fused checkpoint where branches are re-parameterized.

import torch
from mobileone import mobileone

model = mobileone(variant='s0', inference_mode=True)
checkpoint = torch.load('/path/to/checkpoint.pth.tar')
model.load_state_dict(checkpoint)
# ... evaluate/demo ...

ModelBench App

An iOS benchmark app for MobileOne CoreML models. See ModelBench for addition details on building and running the app.

Citation

If our code or models help your work, please cite our paper:

@article{mobileone2022,
  title={An Improved One millisecond Mobile Backbone},
  author={Vasu, Pavan Kumar Anasosalu and Gabriel, James and Zhu, Jeff and Tuzel, Oncel and Ranjan, Anurag},
  journal={arXiv preprint arXiv:2206.04040},
  year={2022}
}

About

This repository contains the official implementation of the research paper, "An Improved One millisecond Mobile Backbone".

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Swift 52.6%
  • Python 46.7%
  • Shell 0.7%