发新帖

TensorflowLite - 如何使用TensorFlowLite在Android App中实现人形检测

[复制链接]
1161 8

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
本帖最后由 rustfisher 于 2018-4-29 15:55 编辑

第一次接触TensorFlow,遇到了很多问题,想向大家求教。如何使用TensorFlowLite在Android App中实现人形检测?

开发环境:
win7, Ubuntu16.04虚拟机,
Python 2.7.12 , TensorFlow 1.7.0 CPU版
已在Ubuntu虚拟机中下载好TensorFlow源码,安装了bazel工具,尝试运行过Toco

运行Google提供的TensorFlowLite Android示例工程,使用mobilenet_quant_v1_224.tflite与labels.txt可以在安卓手机上实时检测出一些物体。
检查labels.txt发现,其中没有person/human/people之类的信息。
而在TensorFlow Android示例中,利用ssd_mobilenet_v1_android_export.pb和coco_labels_list.txt可以检测到人形信息。

接下来想做的是:将ssd_mobilenet_v1_android_export.pb转换为tflite格式,实现Android App 通过 TensorFlowLite 检测到人形信息的功能。尝试过Toco,但一直不成功
已尝试:bazel run --config=opt \
  //tensorflow/contrib/lite/toco:toco -- \
  --input_file=/home/rustfisher/ws/ws_tensorflow/ssd_mobilenet_v1_android_export/ssd_mobilenet_v1_android_export.pb \
  --output_file=/home/rustfisher/ws/ws_tensorflow/ssd_mobilenet_v1_android_export/ssd_mobilenet_v1_android_export.tflite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --inference_type=QUANTIZED_UINT8 \
  --input_shape=1,128,128,3 \
  --input_array=input \
  --output_array=MobilenetV1/Predictions/Reshape_1 \
  --mean_value=128 \
  --std_value=127

报错:
F tensorflow/contrib/lite/toco/tooling_util.cc:722] Check failed: model.HasArray(output_array) Output array not found: MobilenetV1/Predictions/Reshape_1


请大家不吝赐教,

谢谢!



我知道答案 回答被采纳将会获得10 金币 + 10 金币 已有8人回答
本楼点评(1) 收起
  • 阮永兴您好,我和您有同样的需求,请问您实现人形识别的功能了吗?能否交流下,谢谢!
    2018-7-27 15:59 回复

精彩评论8

舟3332  TF芽芽  发表于 2018-4-25 15:55:22 来自手机  | 显示全部楼层
发一下具体的错误信息吧~
本楼点评(1) 收起
rustfisher  TF荚荚  发表于 2018-4-29 15:56:17 | 显示全部楼层
舟3332 发表于 2018-4-25 15:55
发一下具体的错误信息吧~

命令和错误信息已更新
本楼点评(0) 收起
M丶Sulayman  TF豆豆  发表于 2018-4-29 18:15:53 | 显示全部楼层
哇,现在都这么方便了,等我把基础学扎实了,我也看看TFLite
本楼点评(0) 收起
M丶Sulayman  TF豆豆  发表于 2018-4-29 18:17:14 | 显示全部楼层
linux系统上运行和WINDOWS上区别大么?...怎么感觉linux上的都很高大上
本楼点评(0) 收起
舟3332  TF芽芽  发表于 2018-5-2 17:56:25 | 显示全部楼层
M丶Sulayman 发表于 2018-4-29 18:17
linux系统上运行和WINDOWS上区别大么?...怎么感觉linux上的都很高大上

个人感觉 Linux 做开发方便很多哈
本楼点评(0) 收起
舟3332  TF芽芽  发表于 2018-5-2 17:57:47 | 显示全部楼层
rustfisher 发表于 2018-4-29 15:56
命令和错误信息已更新

不好意思~ ssd_mobilenet_v1_android_export 这个模型是在哪里下载的呀?我也去试试~
本楼点评(0) 收起
neverchange  TF豆豆  发表于 2018-7-3 21:48:14 | 显示全部楼层
那个脚本的输出写的不对,输出是模型,你写的目录是什么意思?
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表