TensorFlow与android结合

mind_programmonkey 2018-06-19 08:46
关注文章

1.Android与TensorFlow的结合

在android使用TensorFlow有两种方式:

     1.TensorFlow for mobile,较为成熟,包含的功能方法多

     2.TensorFlow Lite,2017年11月15日,谷歌发布的的TensorFlow Lite的开发者预览版本,这是一款TensorFlow用于设备和嵌入式设备的轻量级方案,允许设备端的机器学习模型的低延迟推断。

2.Android for mobile

TensorFlow提供了一个官方的demo,

TensorFlow官方demo

第一步:在build.gradle里添加依赖compile 'org.tensorflow:tensorflow-android:+'即可

第二步:调用TensorFlow接口并进行使用;

官网的代码:

description

description

3.Android TensorFlow lite

步骤:

  1. 将官方TensorFlow Lite项目导入到Android Studio:

TensorFlow Lite

  1. 下载移动端的模型(model)和标签数据(lables):

移动端模型和数据

  1. 下载完成解压mobilenet_v1_224_android_quant_2017_11_08.zip文件得到一个xxx.tflite和labes.txt文件,分别是模型和标签文件,并且把这两个文件复制到assets文件夹下。

  2. 构建app,run……

在这里我事先的完成了一个物体识别的Demo,下方给出了对影像中物体是什么的判断,前面一个字段是可能的物体名称,后面一个字段是概率。

物体识别这个项目在Android For Tensorflow这个文件夹中tensorflowlite-demo中,tensorflow-master.zip这个压缩包是谷歌官方提供的demo。而mobilenet_android_quant.zip这里面有我们移动端所需要的模型(model)和标签数据(lables)

description

最后给出demo的连接地址:

demo的地址

文章被以下专辑收录
{{panelTitle}}
支持Markdown和数学公式,公式格式:\\(...\\)或\\[...\\]

还没有内容

关注微信公众号