1. 程式人生 > >Alexnet在android手機中的tensorflow和arm compute libraray的效能對比

Alexnet在android手機中的tensorflow和arm compute libraray的效能對比

首先先說明一下這個alexnet是進過修改的網路,由於模型太大,我把alexnet的全連線層6和全連線7給刪除了,測試手機是紅米Note 4X,cpu是高通驍龍625。

分別放出測試程式碼的github
arm compute library:

tensorflow:

測試結果為:
arm compute lilrary:0.422s
tensorflow:0.713s

arm compute library比tensorflow快了差不多百分之40。arm這個也是有一定的優化,但是現在還是有挺多bug,而且介面暫時只提供C++,可以等之後成熟了再使用。