目标检测第1步-运行tensorflow官方示例

2019-01-04 本文已影响22人机器学习算法工程师

前言

本文在学习《Tensorflow object detection API 搭建属于自己的物体识别模型（1）——环境搭建与测试》的基础上优化并总结，此博客链接：https://blog.csdn.net/dy_guox/article/details/79081499，感谢此博客作者。

在进行本文操作之前，需要先安装好tensorflow的gpu版本。

本文作者的环境：python3.6、Windows10、tensorflow_gpu1.10
已经安装好的可以跳过，学习如何安装tensorflow的gpu版本的读者请阅读本文作者的另外一篇文章《深度学习环境搭建-CUDA9.0、cudnn7.3、tensorflow_gpu1.10的安装》，链接：https://www.jianshu.com/p/4ebaa78e0233

本文是写给目标检测入门新手的指导文章，会用示意图将每一步的详细实现过程展示出来。

本文作者接触深度学习2个月后，开始进行目标检测实践。

本文作者的专题《目标检测》，链接：https://www.jianshu.com/c/fd1d6f784c1f

此专题的宗旨是让基础较为薄弱的新手能够顺利实现目标检测，专题内容偏向于掌握技能，学会工具的使用。

本文作者尚未具备清楚讲述目标检测原理的能力，学习原理请自行另找文章。

访问tensorflow在github的主页

tensorflow在github的主页链接：https://github.com/tensorflow
主页界面如下图所示。

Repository中文叫做知识库，对于程序员来说即代码库。

tensorflow这个用户中有很多Repository，我们需要的目标检测代码在models这个Repository中。

如下图2个红色箭头标注处所示，都是我们想要找的models这个代码库。

这2个红色箭头标注处，随意选择一个点击，进入下一步。

注意：因为本文写作的时间与读者浏览网址的时间不同，所以本文写作时的界面与读者浏览网址时的界面也可能不同，读者需要在主页中往下翻即可找到models这个代码库。

image

models这个代码库也可以使用链接直接访问：https://github.com/tensorflow/models

代码库界面如下图所示。

从下图中上方红色箭头标注处可以看出，这个代码库被44617人点赞，说明代码库的流行程度高，受众程度广。

下图中下方红色箭头标注处的research文件夹，我们需要的目标检测在这个文件夹中。

点击下图中下方红色箭头标注处，进入下一步。

image

models这个代码库中的research文件夹也可以使用链接直接访问：https://github.com/tensorflow/models/tree/master/research

models这个代码库中的research文件夹页面如下图所示：

image

从上图中可以看出没有object_detection这个文件夹，读者需要在页面中往下翻可以找到。

在页面中往下翻找到object_detection文件夹如下图红色箭头标注处所示。

点击下图中红色箭头标注处，进入下一步。

image

models这个代码库中的research/object_detection文件夹页面如下图所示。

也可以使用链接直接访问：https://github.com/tensorflow/models/tree/master/research/object_detection

image

在上图页面中往下翻，可以看到research/object_detection文件夹的指导文件README.md，如下图所示。

阅读README.md中的内容，可以获得英文版Tensorflow Object Detection API的使用指导。

image

下载代码库

在models代码库的界面中，有下载整个代码库的按钮，如下图红色箭头标注处所示：

image

通过vpn访问github，速度会加快。

从下图中可以看出，本文作者下载此代码库的速度为5.4MB/s

image

本文作者将此代码库上传到百度云盘，下载链接: https://pan.baidu.com/s/1TnHvqNLT0JEZlugrdyABhQ提取码: fnp3

对于本文的读者来说，其实只需要代码库models中的research/object_detection文件夹就可以。

代码库models文件大小有439M，但是object_detection文件夹只有17M左右。

压缩文件object_detection.zip的百度云盘下载链接：https://pan.baidu.com/s/1Q9SxtKlOqEty08tpFeUUHA提取码: p2sm

下载完成后，在桌面新建文件夹目标检测，将压缩文件object_detection.zip放入其中，如下图所示。

image

选择解压到object_detection，如下图所示。

image

解压完成后，文件夹目标检测中的文件情况如下图所示。

image

proto文件转py文件

(1) 下载和安装protoc

Protobuf(Google Protocol Buffers)是google开发的的一套用于数据存储，网络通信时用于协议编解码的工具库。它和XML和Json数据差不多，把数据已某种形式保存起来。Protobuf相对与XML和Json的不同之处，它是一种二进制的数据格式，具有更高的传输，打包和解包效率。

下载Protobuf网址：https://github.com/google/protobuf/releases

下载Protobuf网址页面如下图所示，下图中红色箭头标注处是Protobuf在操作系统Windows中可以直接运行的protoc程序，下载该压缩文件protoc-3.6.1-win32.zip。