图像识别 tensorflow object detection

我们在使用tensorflow做图像识别的时候，会用到tensorflow object detection api，但是官方没有给出一个详细的安装教程，下面我总结了一个教程，希望能够帮到大家，下面每一个步骤都是必须的，必须按顺序执行，任何一个步骤失败都不行。

下载源代码

可以通过git方式下载，但是很慢，也可以到github直接下载zip

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

安装protobuf

Linux

sudo apt-get install protobuf-compiler

Mac OS 安装 protobuf
到 https://github.com/protocolbuffers/protobuf/releases 下载protoc-3.6.1-osx-x86_64.zip，解压后

cd protoc-3.6.1-osx-x86_64
cp -r include/ /usr/local/include/
cp -r bin/ /usr/local/bin/

编译proto文件

cd research
protoc object_detection/protos/*.proto --python_out=.

安装slim

cd research/slim
python setup.py build
python setup.py install

安装object_detection

cd research
python setup.py build
python setup.py install

测试是否安装完成

cd research
python object_detection/builders/model_builder_test.py

如果执行结果如下，说明安装成功，否则根据结果看看那个步骤出现了问题

......................
----------------------------------------------------------------------
Ran 22 tests in 0.154s

OK

报错汇总

报错一：没有安装object_detection

ModuleNotFoundError: No module named 'object_detection'

报错二：没有编译proto文件

ImportError: cannot import name 'string_int_label_map_pb2'

报错三：没有安装slim

ModuleNotFoundError: No module named 'nets'