![深度学习图像识别技术:基于TensorFlow Object Detection API和OpenVINO? 工具套件](https://wfqqreader-1252317822.image.myqcloud.com/cover/997/38485997/b_38485997.jpg)
2.7 安装TensorFlow Object Detection API框架
2.7.1 TensorFlow Object Detection API简介
TensorFlow Object Detection API是一个在TensorFlow基础上开发出来的用于计算机视觉领域实现在图像中检测并定位多个目标物体的软件框架。
基于TensorFlow Object Detection API开源软件框架,开发者可以快速地创建、训练和部署目标检测模型(Object Detection Model),如图2-47所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/59_02.jpg?sign=1739464994-3ltosCdtI8w9VT1hRzpwauZZ2eZB2KcS-0-2c493dbeff7e64dbe949577d836c5eb8)
图2-47 TensorFlow Object Detection API开源软件框架
2.7.2 下载并安装
TensorFlow Object Detection API的下载安装步骤如下。
第一步,建立一个用于TensorFlow训练的文件夹目录结构,将所有基于TensorFlow训练的相关文件集中到一起,方便文件管理。
请读者新建一个文件夹tf_train,文件夹的名字可以根据自己的命名习惯来取。本书推荐tf_train。tf是TensorFlow的通用简称,train表明该文件夹用于管理训练相关的文件。
在tf_train文件夹下,创建一个名字为addons的文件夹,用于存放附加组件或其他软件工具;再创建一个名字为workspaces的文件夹,用于存放每一个具体项目的文件,如图2-48所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/60_01.jpg?sign=1739464994-QXQu8Wk7HjdSSPYKbEVi1Lms9vxBtqgp-0-725a1e621c642555d422b87f21127729)
图2-48 tf_train文件夹
第二步,在tf_train文件夹中的空白处单击右键,在右键菜单中单击Git Bash Here,启动Git Bash。
由于TensorFlow Object Detection API的master分支不断在更新,为了保证读者能复现本书的操作,本书使用的TensorFlow Object Detection API的版本分支(Branch)是r1.13.0,链接为https://github.com/tensorflow/models/tree/r1.13.0。
在Git Bash中输入git clone命令,并加入参数-b r1.13.0,告诉git克隆r1.13.0分支。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/60_02.jpg?sign=1739464994-CrFVqXrcOkCXA8fDRf9EgAlT6e72K80j-0-4f2acab992986334e856ac0ca1ec8b65)
下载TensorFlow Object Detection API的相关软件源代码,如图2-49所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/60_03.jpg?sign=1739464994-KZYIzSx3P8aCECGZFrwyBVtEFFD7UCRo-0-543a3ec0a2554987d0ea0909d2cfbc52)
图2-49 下载TensorFlow Object Detection API的软件源代码
下载完毕后可以得到一个models文件夹,请把models文件夹全部复制到tf_train文件夹下,这样文件夹结构如图2-50所示,完成TensorFlow Object Detection API的下载和安装。注意:若git下载速度太慢,请从本书提供的360云盘中获取。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/60_04.jpg?sign=1739464994-2sNjXp0GfeLp9WpTqsOW1cCvfBJZa5j9-0-a90185b38f839210ff1d3fedfd9527c6)
图2-50 复制models到tf_train
2.7.3 安装依赖的python软件包
TensorFlow Object Detection API框架依赖的Python软件包见表2-2。
表2-2 依赖软件包及其用途
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/61_01.jpg?sign=1739464994-GK3HZ7Q991mGKzh8hKKzVmMNj5kffusl-0-16d08f5ca74e96271043178fae2660f7)
从Windows“开始”菜单启动Anaconda Navigator,在Environments选项卡处单击tf_gpu,先激活虚拟环境tf_gpu;然后再单击tf_gpu右边的绿色箭头,在弹出菜单中选择Open Terminal,然后在弹出的命令行终端中输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/61_02.jpg?sign=1739464994-ab7WlCz0qGmNaob9q2pUVsZyYcIWhUOp-0-be6902bd9c0615f8354e7a109aeee7b1)
完成TensorFlow Object Detection API框架依赖的Python软件包的安装,运行结果如图2-51所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/61_03.jpg?sign=1739464994-m8lL5dLaxKt4jAPzPoYRaQfmSS42TZTV-0-4a280448cb2da1afca135511fbd867c2)
图2-51 安装依赖软件包
2.7.4 配置环境变量
为了让Python能找到TensorFlow Object Detection API依赖的软件模块,需要给Python的模块搜索路径变量PYTHONPATH添加三个路径,如下所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/62_01.jpg?sign=1739464994-ACnpCVbaUN1EBgSXZ4hRua4co5VYkOzB-0-2cff1fc9528cf87ab745bbafe8001afb)
将<PATH_TO_TF>替换为tf_train文件夹所在的绝对路径,例如,本书中tf_train文件夹所在的绝对路径是D:\,那么就把<PATH_TO_TF>替换为D:\,替换结果如下。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/62_02.jpg?sign=1739464994-JGtkjXEUxnDxDCFIDlV7gi7xQAqeDWmp-0-912ee9decff95d25290c9711c470da75)
通常情况下,Windows的系统变量中是没有PYTHONPATH的,需要用户新建一个,具体步骤如下。
第一步,在Windows任务栏的搜索框中输入“环境变量”,接着在搜索结果中单击“编辑系统环境变量”,启动“环境变量”窗口,如图2-52所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/62_03.jpg?sign=1739464994-Hd77wur42MAFA8dY3xcVe52oXqj569Bq-0-c148a34969ffc72e7b5a8f59c46ef269)
图2-52 启动“环境变量”窗口
第二步,在“环境变量”窗口单击“新建”按钮,然后在弹出的“新建系统变量”窗口的“变量名”输入栏中输入“PYTHONPATH”;在“变量值”输入栏中输入:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/63_01.jpg?sign=1739464994-WomG5ZR6hpMAcIrldQZW2M6KJxdfEc87-0-52ef8cf97470c7efa83242e32d31c03e)
最后单击“确定”按钮,完成环境变量配置工作,如图2-53所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/63_02.jpg?sign=1739464994-pLWh5iih5ldeRyfQQhrv4cJWjhTq8gvv-0-8a5ce917e227369d279ab9dd4c3c32c6)
图2-53 设置PYTHONPATH变量
2.7.5 安装COCO API
TensorFlow Object Detection API默认评估指标是Pascal VOC,若希望使用COCO评估指标,则需要安装pycocotools,具体步骤如下。
第一步,下载并安装Visual C++ 2015编译工具visualcppbuildtools_full.exe,下载链接为https://go.microsoft.com/fwlink/?LinkId=691126。
第二步,在addons文件夹的空白处单击右键,在右键菜单中单击Git Bash Here,启动Git Bash,然后在Git Bash中输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/64_01.jpg?sign=1739464994-hkBUDrMXWV1rMC9dCXUzsNyujq6wZ38w-0-55df6da1a026824a145705851d1b3a49)
下载cocoapi源代码,如图2-54所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/64_02.jpg?sign=1739464994-sdoo1pQPELfcM68VgPRrqygcXNQigHYI-0-e7aed605ca6af34c7a7db4a70a4033ac)
图2-54 下载cocoapi源代码
下载完毕后,文件夹目录结构如图2-55所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/64_03.jpg?sign=1739464994-W7jlS0mFKmqHZJeQqKZVm9RFNwkpEZMy-0-53c85258d96e3abc256aed44b5b14bea)
图2-55 cocoapi的目录结构
第三步,在Windows任务栏的搜索框中输入“Anaconda Prompt(Anacona3)”,在搜索结果中右键单击Anaconda Prompt(Anacona3),在右键菜单中选择“以管理员身份运行”,启动Anaconda命令行终端,如图2-56所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/65_01.jpg?sign=1739464994-LlIuJ2iHykVg53sDTScvlsgVSYEWaciN-0-52a14c349005d4a8c224c50dceac09dc)
图2-56 以管理员身份运行Anaconda Prompt
然后输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/65_02.jpg?sign=1739464994-Zk9c4zEaiqLSBFJuiaPfWzH3gu0yvMsM-0-90e7a8402d7b21002a5fe2993e6d6265)
激活tf_gpu虚拟环境,并输入命令“d:”,进入d盘,再输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/65_03.jpg?sign=1739464994-FaAv9f33q6QwuFT13AFERDZmXS3e4huV-0-1f8f9bba575387be0002c2ef9b1efead)
进入PythonAPI文件夹,最后输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/65_04.jpg?sign=1739464994-BZKF8ipvk62qL56gaBg1QGVKq80H2Dx4-0-7a28b0da76d7523e1b16c290f4da3ae9)
完成COCO API的安装,如图2-57所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/65_05.jpg?sign=1739464994-qzCSYTP723Nrf7L8nw2XLzIDfoaYt3QK-0-03a91ab0dbb5504c743920cd08f62ec8)
图2-57 安装COCO API完成
第四步,在安装好COCO API的命令行终端里输入命令“python”,启动Python环境,然后输入语句“import pycocotools”,若import成功,则说明COCO API安装成功,如图2-58所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/66_01.jpg?sign=1739464994-r76W51wMqD3FlnRcwC7Db6GQiQupVJhZ-0-9d4c1da93c42fba1effe975dec4046c6)
图2-58 import pycocotools成功
2.7.6 编译proto文件
Tensorflow Object Detection API框架使用protobuf工具来配置模型和训练参数。protobuf工具在2.5.2节用命令conda install tensorflow-gpu=1.13.1安装TensorFlow时,被conda附带安装过了,可以用命令conda list查看确认。
在运行Tensorflow Object Detection API框架前,必须编译proto文件。请进入models\research文件夹(D:\tf_train\models\research),在文件夹地址栏中输入“cmd”,启动Windows命令行终端,然后输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/66_02.jpg?sign=1739464994-9Yvx6Uc9YBHeoIeM9Fxv8mYrBA6dWjLR-0-38af18efe5d821bdc1bbcc1bf690b916)
激活tf_gpu虚拟环境,最后输入命令:
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/66_03.jpg?sign=1739464994-KyBoETVrUykLkDXG0bRdUDWExSA6prZC-0-cae60fb8f76e58ff7fffa3c485483ebc)
完成对proto文件的编译,如图2-59所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/66_04.jpg?sign=1739464994-BvTnRhu2fa2SWFlzrz4rx45EpOqbOarc-0-34478bb46863ad470bfd01daf7d4f89a)
图2-59 编译proto文件
2.7.7 测试安装
至此,TensorFlow Object Detection API框架及其依赖项就安装完毕了,接下来的工作是测试其是否已经安装成功,具体步骤如下。
第一步,从TensorFlow预训练模型库
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/67_01.jpg?sign=1739464994-Bp22RYp84wMSXX5oP5eWETWryN3jpIGJ-0-773f726089295b85f44d11a7e4ebbf1c)
中下载模型ssd_inception_v2_coco,如图2-60所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/67_02.jpg?sign=1739464994-HzRxcQClsVpIHLHZ5tWmAl6FqpQnFISj-0-b880df59372189d7affa3f5e56bc5357)
图2-60 ssd_inception_v2_coco
第二步,模型下载后,得到文件ssd_inception_v2_coco_2018_01_28.tar.tar,直接将其解压,如图2-61所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/67_03.jpg?sign=1739464994-hrx62sGT3ezsB9mIaN8a9dXtR6VDNddX-0-44f0168e0dcf31817657e374a6faa855)
图2-61 解压ssd_inception_v2_coco_2018_01_28.tar.tar
第三步,将解压缩出来的文件夹ssd_inception_v2_coco_2018_01_28复制到D:\tf_train\models\research\object_detection文件夹中,如图2-62所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/68_01.jpg?sign=1739464994-FHEqTL9yjggNItXHtdOZUaTwaUDGi5mM-0-b21a79d9e801283424d4b6a27597e51b)
图2-62 复制文件夹ssd_inception_v2_coco_2018_01_28
第四步,将本书附带的object_detection_example_1.py(从本书提供的360云盘中获取)复制到object_detection文件夹。
第五步,修改D:\tf_train\models\research\object_detection\utils文件夹中的visualization_utils.py文件。请将visualization_utils.py文件的第26行import matplotlib;matplotlib.use ('Agg') 注释掉,如图2-63所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/68_02.jpg?sign=1739464994-zt5UhUsZWVfbY63Ye9C8hI3fyNk5ODqs-0-5d1da8b125a078fb5f49c7bf5c06d26a)
图2-63 注释掉import matplotlib;matplotlib.use ('Agg')
第六步,进入object_detection文件夹,在文件夹地址栏中输入“cmd”,启动Windows命令行终端,然后输入命令“conda activate tf_gpu”,激活tf_gpu虚拟环境,最后输入命令“python object_detection_example_1.py”,如图2-64所示。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/69_01.jpg?sign=1739464994-3BFXInLLYTLH71HA4Mzh4mnxJsUEuyle-0-38639891731d77eac65f5f4918dbc5e2)
图2-64 运行object_detection_example_1.py
若弹出如图2-65所示两张识别成功的图片,说明TensorFlow Object Detection API框架及其依赖项已经安装成功,可以支持训练自己的目标检测模型了。
![](https://epubservercos.yuewen.com/31903E/20255828801879906/epubprivate/OEBPS/Images/69_02.jpg?sign=1739464994-TIu2MJxvCChU2OQwJ1zGZoCiH0s5h1TM-0-aa85004ece2d9ec298f1a6c0fdecd58e)
图2-65 object_detection_example_1.py运行结果