博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
如何使用tensorflow for mobile,开发环境为android studio
阅读量:5337 次
发布时间:2019-06-15

本文共 1664 字,大约阅读时间需要 5 分钟。

目前要做一个基于图片识别的安卓app,撇开ui的部分,首先要做的就是在android上把tensorflow跑起来。

在android上使用tensorflow有两种方式:

  1. tensorflow for mobile,较为成熟,包含的功能方法多。
  2. tensorflow lite,是1的升级版,目前处于开发者预览阶段,优势是体积小性能有优化。是未来的趋势。

鉴于项目原因,用的第一种。

第一步,在android studio里添加tensorflow的library引用。

鉴于网络没问题,所以我直接使用第一种方式(Include the jcenter AAR which contains it):
build.gradle里添加依赖compile 'org.tensorflow:tensorflow-android:+'即可

第二步,调用tensorflow接口进行使用。

官网的代码:

// Load the model from disk.TensorFlowInferenceInterface inferenceInterface =new TensorFlowInferenceInterface(assetManager, modelFilename);// 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);

根本看不懂这些参数要怎么设定,不过可以用官方的example,所以就直接copy了图片识别的code

拷贝这两个文件就可以了:和

第三步,进行识别。

// 用model创建一个分类器。final Classifier classifier = TensorFlowImageClassifier.create(       getAssets(),       MODEL_FILE,       LABEL_FILE,       INPUT_SIZE,       IMAGE_MEAN,       IMAGE_STD,       INPUT_NAME,       OUTPUT_NAME);// 加载图片final Bitmap bitmap = BitmapFactory.decodeResource(getResources(), R.drawable.noodle);// 识别图片btn.setOnClickListener(new View.OnClickListener() {   @Override   public void onClick(View view) {       List
results = classifier.recognizeImage(bitmap); for(Classifier.Recognition result : results) { tv.setText(tv.getText().toString() + "\r\n" + result.getTitle()); } }});

至此,成功在android手机跑起了tensorflow的库,真的是很简单好用。

PS:图片的部分遇到arrayOutOfIndex问题就是了。

转载于:https://www.cnblogs.com/happywu/p/8302672.html

你可能感兴趣的文章
Faster_RCNN 2.模型准备(上)
查看>>
ANDROID 2.0游戏开发实战宝典pdf
查看>>
《Android 4高级编程(第3版)》(完整书签).pdf
查看>>
pku 2195 Going Home 最小费最大流问题
查看>>
第四届ACM_DIY群程序设计竞赛 (部分解题报告) 弱菜在此大牛无视。。。
查看>>
通货膨胀 通货紧缩 贸易逆差
查看>>
第四章 Spring.Net 如何管理您的类___让对象了解自己的容器
查看>>
cocos creator制作微信小游戏
查看>>
JavaScript基础函数---李老师的
查看>>
Centos7之Nginx
查看>>
hdu 2121 , hdu 4009 无定根最小树形图
查看>>
Openwrt 安装软件到U盘或硬盘
查看>>
JDK部署
查看>>
Java循环结构第二讲
查看>>
Docker简介
查看>>
【BZOJ 4709】柠檬 斜率优化dp+单调栈
查看>>
Android Studio插件GsonFormat
查看>>
DS博客作业06--图
查看>>
python:requests模块
查看>>
js dom操作基本单词和格式
查看>>