Tensorflew的Android Demo跑起来

本帖子旨在帮助各位初学者填坑,快速运行起来,排版不优美的地方以及有错误的地方请指正。

1. 下载源码
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git checkout r1.2
2. 先测试下用brazel编译

这一步骤可以不做,一基友没做的情况下再Android Studio里面也可以运行成功。

2.1 首先要修改下WORKSPACE文件里的内容,注意下当前的位置
vim tensorflow/WORKSPACE

修改文件中对应行

# 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 = "25.0.2",
    # Replace with path to Android SDK on your system
    path = "/Users/superchange/Library/Android/sdk",
)
#
# Android NDK r12b is recommended (higher may cause issues with Bazel)
# 请不要无视上面这行话,由于我无视了它,导致我多花了1天时间,这是没有必要的,请去下载一个r12的NDK,然后对应着修改下面的路径
android_ndk_repository(
    name="androidndk",
    path="/Users/superchange/Documents/TensorFlow_learn/android-ndk-r12b",
    # This needs to be 14 or higher to compile TensorFlow.
    # Note that the NDK version is not the API level.
    api_level=14)

2.2 运行下面这行命令进行编译
bazel build -c opt //tensorflow/examples/android:tensorflow_demo

编译需要很久,且中间会去谷歌的服务器上下载些文件,所以要保持全局翻墙,否则很慢,还可能会失败,编译成功的提示

CONFLICT: asset:WORKSPACE is provided with ambiguous priority from:
    external/mobile_multibox/WORKSPACE
    external/inception5h/WORKSPACE
CONFLICT: asset:WORKSPACE is provided with ambiguous priority from:
    external/stylize/WORKSPACE
    external/mobile_multibox/WORKSPACE
Target //tensorflow/examples/android:tensorflow_demo up-to-date:
  bazel-bin/tensorflow/examples/android/tensorflow_demo_deploy.jar
  bazel-bin/tensorflow/examples/android/tensorflow_demo_unsigned.apk
  bazel-bin/tensorflow/examples/android/tensorflow_demo.apk
INFO: Elapsed time: 1779.182s, Critical Path: 90.91s

注: CONFLICT应该没有啥关系,我测试了apk可以安装并运行;
我的电脑配置是16年的mbp15寸高配,编译花了我30分钟,编译的时候CPU占用基本上在30%左右,可能主要是下载文件花了时间

3. 使用Android Studio编译并运行
3.1 导入工程到Android

Studio


import.jpeg
3.2 运行demo
run demo.jpeg
4. 训练自己的分类器

本步骤可以实现让app识别自己想让它识别的东西,比如hotdog啥的

4.1 新建文件夹

需要注意的是,至少要准备2个类别,否则训练是不能完成的,因此先新建文件夹,这里我用的是马里奥和皮卡丘

mkdir tf_files
cd tf_files
mkdir games
cd games
mkdir pikaqiu
mkdir mario
cd ../..

现在该去下载图片了,因为只是试验demo行不行,我没有用爬虫去爬,直接用的chrome的插件来批量下载图片
安装插件以后用Google或者百度搜索图片,输入关键字 皮卡丘或者mario

save mario 01.png
save mario 02.jpeg

文件下载完成后去download文件夹里面找到,批量修改图片的名字,再复制的之前新建的文件夹中,每种类别至少30张图片。

4.2 训练模型
python3 tensorflow/examples/image_retraining/retrain.py \
  --bottleneck_dir=/Users/superchange/Documents/TensorFlow_learn/tensorflow/tf_files/bottleneck \
  --how_many_training_steps 4000 \
  --model_dir=/Users/superchange/Documents/TensorFlow_learn/tensorflow/tf_files/inception \
  --output_graph=/Users/superchange/Documents/TensorFlow_learn/tensorflow/tf_files/retrained_graph.pb \
  --output_labels=/Users/superchange/Documents/TensorFlow_learn/tensorflow/tf_files/retrained_labels.txt \
  --image_dir /Users/superchange/Documents/TensorFlow_learn/tensorflow/tf_files/games

注意: 以上代码段中的/Users/superchange/Documents/TensorFlow_learn/路径替换成你自己的对应的路径

4.3 模型优化
python3 tensorflow/python/tools/optimize_for_inference.py \
  --input=/Users/superchange/Documents/TensorFlow_learn/tensorflow//tf_files/retrained_graph.pb \
  --output=/Users/superchange/Documents/TensorFlow_learn/tensorflow//tf_files/optimized_graph.pb \
  --input_names="Mul" \
  --output_names="final_result"

注意: 以上代码段中的/Users/superchange/Documents/TensorFlow_learn/路径替换成你自己的对应的路径

4.4 训练完成

tf_files文件夹下会出现retrained_labels.txtoptimized_graph.pb这两个文件,需要将其复制到/Users/superchange/Documents/TensorFlow_learn/tensorflow/tensorflow/examples/android

5. 修改ClassifierActivity.java中的参数

注意android studio里面要做适当的修改

  private static final int INPUT_SIZE = 299;
  private static final int IMAGE_MEAN = 128;
  private static final float IMAGE_STD = 128f;
  private static final String INPUT_NAME = "Mul";
  private static final String OUTPUT_NAME = "final_result";

  private static final String MODEL_FILE = "file:///android_asset/optimized_graph.pb";
  private static final String LABEL_FILE =
      "file:///android_asset/retrained_labels.txt";

上面的参数都要修改,我尝试过只改文件路径以及INPUT_NAMEOUTPUT_NAME结果程序运行的时候会报错

感谢:
· 感谢大厂google
· 本文主要参考了Nilhcem的帖子

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

推荐阅读更多精彩内容