9 月 25 日,百度在 GitHub 开源了移动端深度学习框架 mobile-deep-learning(MDL)的全部代码以及脚本。
移动端深度学习框架 MDL
授权协议:
MIT
开发语言:
C/C++
操作系统:
跨平台
开发厂商:
百度
Github:
https://github.com/baidu/mobile-deep-learning
mobile-deep-learning(MDL)是一个基于卷积神经网络实现的移动端框架。致力于让卷积神经网络极度简单的部署在手机端。目前正在手机百度内运行。支持 iOS gpu 计算。体积小,速度快。
mobile-deep-learning(MDL)特性如下:
● 一键部署,脚本参数就可以切换 iOS 或者安卓
● 支持 iOS gpu 运行 MobileNet,squeezenet 模型
● 已经测试过可以稳定运行 MobileNet,GoogLeNet v1,squeezenet 模型
● 体积极小,无任何第三方依赖。纯手工打造。
● 提供量化脚本,对 32 位浮点数转 8 位 UINT 直接支持,模型体积量化后 4M 上下
● 与 ARM 相关算法团队线上线下多次沟通,针对ARM平台会持续优化
● NEON 使用涵盖了卷积,归一化,池化所有方面的操作
● 汇编优化,针对寄存器汇编操作具体优化
● 循环展开,为提升性能减少不必要的 CPU 消耗,全部展开判断操作
● 将大量繁重的计算任务前置到开销过程