TensorFlow on Android(6): 看起来像个机器学习专家

2024-02-10 20:48

本文主要是介绍TensorFlow on Android(6): 看起来像个机器学习专家,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

回顾

先来回顾一下我们为了实现一个物体识别器都做了哪些工作:

  1. 找到Pre-traind Model
  2. 找到TensorFlow的Android库
  3. 找到相应的Input Op
  4. 向Input Op 输入相应格式的数据
  5. 找到相应的Output Op
  6. 从Output Op中提取数据并解析

是不是有一种很熟悉的感觉, 这个像极了我们使用一个Jar包或者一个Ruby Gem:先下载相应的库, 然后再看文档知道API的输入参数是什么, 返回结果什么, 然后再调用它。 这个类比的确很像, 不幸的是, 大部分的模型,或者说大部分TensorFlow Models项目里面的模型,并没有这么详细的,标准的文档,有的还需要你自己去导出.pb文件。 如果你不满足于只实现一个物体识别器, 还想用一些其他的模型来解决你的问题, 那么可能你需要自己去找到Inupt/Output Op, 输入参数的格式等。 我们这里仍然以Object Detection API为例子,来看一下可以从哪些地方获取这样的信息。

准备工作##

在开始之前,我们需要安装一些工具, 这些工具也是机器学习专家经常使用的。(COOL! 我们首先学会了一些机器学习的术语, 现在又开始学习使用机器学习的工具了, 下次化妆舞会上你就可以扮演机器学习专家了,没人认得出来!)

Python: 说到机器学习, Python肯定是必不可少的,我也建议大家, 如果不会的话可以先开始学习,在后面的文章中我们会用到的。

在这里可以找到详细的文档,教你在各个平台上面安装Python。

Jupyter Notebook: 这是一个交互式的文档, 可以将文本, 公式, 代码和一些可视化的内容集合到一个文档里面,在机器学习和数据处理的领域里面,你会经常用到它, 安装很简单:

pip install --upgrade pip
pip install jupyter

TensorFlow: 最终我们还是需要安装这个框架,因为我们可能需要写一点Python代码, 在后面的文章中,我们也需要用到。 可以在这里找到相关文档

安装完这3个东西以后,我们就可以正式开始了。

从代码中寻找

代码永远是最好的文档。 TensorFlow Models项目中大部分的模型都有展示如何Inference的Python代码, 读懂这些代码能获取到大量关于如何使用这个模型的信息, 之后你就可以进行其他语言的移植了。

Object Detection API提供了一个Jupyter Notebook的文档, 我们来看一下。
首先你要下载TensorFlow Models的代码

git clone https://github.com/tensorflow/models.git

然后运行Jupyter Notebook

cd models/research/object_detection
jupyter notebook

如果不发生意外的话,命令行终端上应该会打出启动成功的信息, 默认情况下会在localhost:8888上启用一个web服务,并自动打开你的浏览器跳转到相应的页面:
enter image description here

我们点击object_detection_tutorial_ipynb文件,就可以打开这个notebook了
在上面我们可以看到使用这个模型的Python示例代码:

enter image description here

如果你对Python足够熟悉的话, 可以从这些代码里面获取到相应的信息,然后可以根据这些信息再移植到其他语言。

注意这里python代码和我们的Android代码稍微有点不一样,比如

image_tensor = detection_graph.get_tensor_by_name('image_tensor:0')

Python接口里面是通过获取Op的output tensor, 然后在tensor上面操作进行数据的输入和输出,所以这里用’image_tensor:0’ 代表 ‘image_tensor’ Op的第一个output tensor, 这里Java和Python接口的形式不一样, 但是逻辑是一致的。

用TensorBoard来检视模型

TensorBoard是TensorFlow自带的一个可视化工具

如果说从代码里面获取的信息还不够的话, 我们还可以用TensorBoard来直接查看模型的结构,或者用TensorFlow的术语来说,模型的Graph

TensorBoard运行的时候会在本地启动一个Web服务器,然后将相关的数据显示在网页上,我们可以把.pb文件导入到TensorBoard里面,然后显示整个网络的结构。

也许在一个完美的世界里面,我们可以选择文件->打开->选择.pb文件,就可以看到结果了,但是在这个残酷的现实世界里,我们还需要写一点Python代码才能做到这点(已经有人在TensorFlow的项目里面提了这个Feature request, 不过就本博客编写的时候,这个功能还没有实现)

现在找到我们的.pb文件, 然后在同一目录里面创建一个load_graph.py文件, 内容为:

import tensorflow as tf
from tensorflow.python.platform import gfile
with tf.Session() as sess:model_filename ='FILENAME_OF_PB_FILE'with gfile.FastGFile(model_filename, 'rb') as f:graph_def = tf.GraphDef()graph_def.ParseFromString(f.read())g_in = tf.import_graph_def(graph_def)
LOGDIR='log'
train_writer = tf.summary.FileWriter(LOGDIR)
train_writer.add_graph(sess.graph)%

注意你需要将FILENAME_OF_PB_FILE替换为.pb文件的文件名,比如model.pb

然后运行

python load_graph.py

如果没有出现意外的话, 运行这段Python代码之后,会在log文件夹里面生成TensorBoard需要的文件,现在整个目录看起来是这样的
enter image description here

然后我们CD到这个目录, 运行

tensorboard --logdir ./log

然后打开

http://localhost:6006/#graphs

enter image description here

然后双击"import", 整个网络的结构就出来了
enter image description here

在这个图上我们不难找到Input和output Op
enter image description here

enter image description here

我们再观察连接这些Op的箭头上的内容,比如image_tensor上面写着 1x?x?x3, 这代表着这个输入/输出的数据的shape。在这里image_tensor输出给下一个节点的数组Shape为 (1, image_height, image_width, 3)

数组Shape的定义,你可以参考NumPy的相关文档,我们这里不做深入的讨论。 你只需要知道,这个shape和我们在java接口里面调用Feed接口的参数是一致的:

inferenceInterface.feed("image_tensor", byteInput, 1, 300, 300, 3);

一般来说,结合示例代码和网络结构图,如何使用一个模型就应该很清楚了。

一些资源##

  • Caffe Model Zoo
  • Keras Model Zoo
  • Run TensorFlow in your browser
  • Run Keras in your browser

这篇关于TensorFlow on Android(6): 看起来像个机器学习专家的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/698045

相关文章

HarmonyOS学习(七)——UI(五)常用布局总结

自适应布局 1.1、线性布局(LinearLayout) 通过线性容器Row和Column实现线性布局。Column容器内的子组件按照垂直方向排列,Row组件中的子组件按照水平方向排列。 属性说明space通过space参数设置主轴上子组件的间距,达到各子组件在排列上的等间距效果alignItems设置子组件在交叉轴上的对齐方式,且在各类尺寸屏幕上表现一致,其中交叉轴为垂直时,取值为Vert

Ilya-AI分享的他在OpenAI学习到的15个提示工程技巧

Ilya(不是本人,claude AI)在社交媒体上分享了他在OpenAI学习到的15个Prompt撰写技巧。 以下是详细的内容: 提示精确化:在编写提示时,力求表达清晰准确。清楚地阐述任务需求和概念定义至关重要。例:不用"分析文本",而用"判断这段话的情感倾向:积极、消极还是中性"。 快速迭代:善于快速连续调整提示。熟练的提示工程师能够灵活地进行多轮优化。例:从"总结文章"到"用

【前端学习】AntV G6-08 深入图形与图形分组、自定义节点、节点动画(下)

【课程链接】 AntV G6:深入图形与图形分组、自定义节点、节点动画(下)_哔哩哔哩_bilibili 本章十吾老师讲解了一个复杂的自定义节点中,应该怎样去计算和绘制图形,如何给一个图形制作不间断的动画,以及在鼠标事件之后产生动画。(有点难,需要好好理解) <!DOCTYPE html><html><head><meta charset="UTF-8"><title>06

学习hash总结

2014/1/29/   最近刚开始学hash,名字很陌生,但是hash的思想却很熟悉,以前早就做过此类的题,但是不知道这就是hash思想而已,说白了hash就是一个映射,往往灵活利用数组的下标来实现算法,hash的作用:1、判重;2、统计次数;

Android实现任意版本设置默认的锁屏壁纸和桌面壁纸(两张壁纸可不一致)

客户有些需求需要设置默认壁纸和锁屏壁纸  在默认情况下 这两个壁纸是相同的  如果需要默认的锁屏壁纸和桌面壁纸不一样 需要额外修改 Android13实现 替换默认桌面壁纸: 将图片文件替换frameworks/base/core/res/res/drawable-nodpi/default_wallpaper.*  (注意不能是bmp格式) 替换默认锁屏壁纸: 将图片资源放入vendo

零基础学习Redis(10) -- zset类型命令使用

zset是有序集合,内部除了存储元素外,还会存储一个score,存储在zset中的元素会按照score的大小升序排列,不同元素的score可以重复,score相同的元素会按照元素的字典序排列。 1. zset常用命令 1.1 zadd  zadd key [NX | XX] [GT | LT]   [CH] [INCR] score member [score member ...]

Android平台播放RTSP流的几种方案探究(VLC VS ExoPlayer VS SmartPlayer)

技术背景 好多开发者需要遴选Android平台RTSP直播播放器的时候,不知道如何选的好,本文针对常用的方案,做个大概的说明: 1. 使用VLC for Android VLC Media Player(VLC多媒体播放器),最初命名为VideoLAN客户端,是VideoLAN品牌产品,是VideoLAN计划的多媒体播放器。它支持众多音频与视频解码器及文件格式,并支持DVD影音光盘,VCD影

【机器学习】高斯过程的基本概念和应用领域以及在python中的实例

引言 高斯过程(Gaussian Process,简称GP)是一种概率模型,用于描述一组随机变量的联合概率分布,其中任何一个有限维度的子集都具有高斯分布 文章目录 引言一、高斯过程1.1 基本定义1.1.1 随机过程1.1.2 高斯分布 1.2 高斯过程的特性1.2.1 联合高斯性1.2.2 均值函数1.2.3 协方差函数(或核函数) 1.3 核函数1.4 高斯过程回归(Gauss

【学习笔记】 陈强-机器学习-Python-Ch15 人工神经网络(1)sklearn

系列文章目录 监督学习:参数方法 【学习笔记】 陈强-机器学习-Python-Ch4 线性回归 【学习笔记】 陈强-机器学习-Python-Ch5 逻辑回归 【课后题练习】 陈强-机器学习-Python-Ch5 逻辑回归(SAheart.csv) 【学习笔记】 陈强-机器学习-Python-Ch6 多项逻辑回归 【学习笔记 及 课后题练习】 陈强-机器学习-Python-Ch7 判别分析 【学

系统架构师考试学习笔记第三篇——架构设计高级知识(20)通信系统架构设计理论与实践

本章知识考点:         第20课时主要学习通信系统架构设计的理论和工作中的实践。根据新版考试大纲,本课时知识点会涉及案例分析题(25分),而在历年考试中,案例题对该部分内容的考查并不多,虽在综合知识选择题目中经常考查,但分值也不高。本课时内容侧重于对知识点的记忆和理解,按照以往的出题规律,通信系统架构设计基础知识点多来源于教材内的基础网络设备、网络架构和教材外最新时事热点技术。本课时知识