人工智能 - TensorFlow Android 版 [7]

欢迎Follow我的GitHub,关注我的简书

在Android项目中,集成TensorFlow,目前(170920)TensorFlow 1.3版本的TF Detect模块无法使用,观察效果,需要使用1.2版本。

TF Android

项目位置:tensorflow/tree/master/tensorflow/examples/android

配置

下载Homebrew

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"

Bazel用于构建项目,必须要安装。最好的方式是通过Brew下载,由于Bazel较大,下载较慢,经常失败,每次从断点处自动连接,需要不断重复下载,直至全部下载完成。

brew install bazel

在TensorFlow的根目录下,创建download_android_model.sh,添加

BASE_URL=https://storage.googleapis.com/download.tensorflow.org/models
for MODEL_ZIP in inception5h.zip ssd_mobilenet_v1_android_export.zip stylize_v1.zip
do
  curl -L ${BASE_URL}/${MODEL_ZIP} -o /tmp/${MODEL_ZIP}
  unzip /tmp/${MODEL_ZIP} -d tensorflow/examples/android/assets/
done

用于下载Android的模型,执行

sh ./download_android_model.sh

配置文件,修改根目录的WORKSPACE,打开Android相关的脚本,填写androidsdk路径和androidndk路径,并且修改build_tools_version为26.0.1,支持Bazel编译。

# Uncomment and update the paths in these entries to build the Android demo.
android_sdk_repository(
    name = "androidsdk",
    api_level = 23,
    # Ensure that you have the build_tools_version below installed in the
    # SDK manager as it updates periodically.
    build_tools_version = "26.0.1",
    # Replace with path to Android SDK on your system
    path = "/Users/wangchenlong/Installations/android-sdk",
)

# Android NDK r12b is recommended (higher may cause issues with Bazel)
android_ndk_repository(
    name="androidndk",
    path="/Users/wangchenlong/Installations/android-ndk-r10e",
    # This needs to be 14 or higher to compile TensorFlow.
    # Please specify API level to >= 21 to build for 64-bit
    # archtectures or the Android NDK will automatically select biggest
    # API level that it supports without notice.
    # Note that the NDK version is not the API level.
    api_level=14)

根目录的BUILD文件注释assets的external_assets选项,使用本地asset的模型文件。

android_binary(
    assets = [
        "//tensorflow/examples/android/assets:asset_files",
        # ":external_assets",
    ],
)

参考:
bazelasset conflict


构建

在项目的根目录下,使用Bazel构建Android项目,执行项目中的BUILD文件,第一次构建时间较长,耐心等待。

bazel build -c opt //tensorflow/examples/android:tensorflow_demo

成功,没有任何错误。

Bazel

安装Android APK,生成三个入口:TF Classify、TF Detect、TF Stylize。

adb install -r bazel-bin/tensorflow/examples/android/tensorflow_demo.apk

AS

本地执行

gradle assembleDebug

全部编译通过,buildNativeBazel较慢,耐心等待。

点击Android Studio的Run按钮,即可。

如果是小米手机,like me,在调度程序时提示“Installation failed with message Failed to establish session”错误,需要在在开发者选项里关闭MIUI优化。

如果出现错误:

java.io.FileNotFoundException: .gradle/buildOutputCleanup/cache.properties (No such file or directory)

则删除.gradle文件夹,即可,参考


原理

tensorflow的Java包,来源于Bazel的构建,

compile 'org.tensorflow:tensorflow-android:+'

核心类是Classifier,调用接口类TensorFlowInferenceInterface,执行常见的TF操作,如feed、run、fetch等操作。

// Copy the input data into TensorFlow.
inferenceInterface.feed(inputName, floatValues, 1, inputSize, inputSize, 3);

// Run the inference call.
inferenceInterface.run(outputNames, logStats);

// Copy the output Tensor back into the output array.
inferenceInterface.fetch(outputName, outputs);

模型数据存储于asset文件夹中,文件大小约

tensorflow_inception_graph.pb  53.9MB
ssd_mobilenet_v1_android_export.pb  29.1MB
stylize_quantized.pb  564KB

效果

Demo

Thanks 我的同学 @高岩

OK, that's all!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,445评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,889评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,047评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,760评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,745评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,638评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,011评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,669评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,923评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,406评论 4 320
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,995评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2 342

推荐阅读更多精彩内容