mobilenet

mobilenet 在速度、模型大小上做了优化,并保持精度基本不变。



mobilenet 采用了depthwise separable convolutions(L. Sifre. Rigid-motion scattering for image classification, 2014. 1, 3) 的思想,在用3x3(或更大尺寸)卷积的时候并不对通道进行融合,而是采用depthwise(或叫channelwise)和1x1 pointwise的方法进行分解卷积。如下图


计算量从降低为


整个mobilenet的架构如下图


为了更好地控制计算量,mobilenet引入Width Multiplier和Resolution Multiplier分别对网络进行瘦身和降低分辨率。

在图片分类任务上其大致的效果如下



mobilenet_v1.png

在更多任务上的比较见作者文章: https://arxiv.org/pdf/1704.04861.pdf

有tensorflow的实现: https://github.com/tensorflow/models/blob/master/slim/nets/mobilenet_v1.md

caffe也有人实现: https://github.com/shicai/MobileNet-Caffe, 他是通过caffe 的group参数来实现channelwise的操作的,由于实现的问题和cuda/cudnn对其支持得不好,训练起来十分慢。前向预测时在CPU上的耗时大概是googlenet的70%。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值