Alexnet在android手机中的tensorflow和arm compute libraray的性能对比

来源:互联网 发布:windows nginx 子域名 编辑:程序博客网 时间:2024/06/05 19:26

首先先说明一下这个alexnet是进过修改的网络,由于模型太大,我把alexnet的全连接层6和全连接7给删除了,测试手机是红米Note 4X,cpu是高通骁龙625。

分别放出测试代码的github
arm compute library:

https://github.com/ppplinday/alexnet-armcomputelibrary-android-withoutfc67

tensorflow:

https://github.com/ppplinday/alexnet-tensorflow-android-withoutfc67

测试结果为:
arm compute lilrary:0.422s
tensorflow:0.713s

arm compute library比tensorflow快了差不多百分之40。arm这个也是有一定的优化,但是现在还是有挺多bug,而且接口暂时只提供C++,可以等之后成熟了再使用。

原创粉丝点击