Android端高性能圖像分類解決方案

  因爲公司業務須要,前段時間開始瞭解AI方面的東西,準備找一個在android端性能較高的前向計算框架,瞭解了tflite,百度的mdl和騰訊的ncnn,最終敲定ncnn,不失所望,效果很不錯,基本達到預期目標,大致流程以下:

1.選擇適用的網絡結構,我選的是shufflenet(由於是移動端,性能爲上)

2.使用caffe訓練相應網絡對應的模型(煉丹,此步驟坑比較多,對於AI門外漢來講只能作一些調參,基本動不了網絡結構)

3.將caffe訓練好的模型轉換爲ncnn支持的模型(ncnn有相應轉換工具和教程)

4.編寫jni,調用轉換好的模型進行預測

 

不得不說,ncnn真不愧爲衆多開源框架中的最良心的了,教程完備,更新快,速度快