1. 程式人生 > >在Android裝置上配置TensorFlow(二)使用Android Studio呼叫TF

在Android裝置上配置TensorFlow(二)使用Android Studio呼叫TF

根據官方文件的介紹,最簡單的修改,編譯並且執行演示應用的方式是通過Android Studio。雖然官方文件強調通過原始碼編譯中涉及到的Bazel元件並不支援windows, 但是為了便於操作,除非涉及到通過原始碼編譯,我們依然使用windows進行操作。在本文中,我們使用Android Studio開啟樣例工程。

首先用Git Clone整個專案:

git clone --recurse-submodules https://github.com/tensorflow/tensorflow.git
用Android Studio開啟專案所在的project, 如果遇到版本相容問題,按照Android Studio的提示解決即可。
D:\tensorflow\tensorflow\examples\android

由於Windows下不支援使用Bazel從原始碼編譯,我們也不需要編譯TensorFlow的原始碼,因此我們把它關掉。修改build.gradle檔案,把nativeBuildSystem的值改成none。
def nativeBuildSystem = 'none'

這樣我們就可以修改原始碼,編譯執行啦。TensorFlow現在會把咖啡杯識別成別人。


參考文獻:

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android/