请选择 进入手机版 | 继续访问电脑版
  • 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    迪恩网络公众号

apple/ml-mobileone: This repository contains the official implementation of the ...

原作者: [db:作者] 来自: 网络 收藏 邀请

开源软件名称(OpenSource Name):

apple/ml-mobileone

开源软件地址(OpenSource Url):

https://github.com/apple/ml-mobileone

开源编程语言(OpenSource Language):

Swift 52.6%

开源软件介绍(OpenSource Introduction):

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}
}



鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap