ubuntu16.04下Anaconda3环境中安装TensorFlow object Detection API

我的配置：Anaconda3 5.2.0、tensorflow-gpu1.9、cuda9、cuDNN7，以上具体安装过程可参考我以前的博文。下面具体讲一下 我在Anaconda3环境中安装TensorFlow object Detection API的过程：

1、TensorFlow Model模型库下载

下载链接：https://github.com/tensorflow/models

下载完成后解压，将文件重命名为models，并保把该文件保存在你所安装的tensorflow路径中(如何查询tensorflow的路径请参考我以前的博文)，我的保存路径如下图，其中models文件夹已放入其中。

2、protobuf编译

2.1 利用Anaconda安装protobuf

2.2 编译proto文件

**安装tensorflow的anaconda环境，进入models文件夹下的research文件夹中，输入如下命令(见下图)：

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

2.3 运行完成后，可以检查/home/lg/anaconda3/envs/tensorflow-gpu/lib/python3.6/site-packages/tensorflow/models/research/object_detection/protos/文件夹(注:此处是我的文件路径)，如果每个proto文件都生成了对应的.py文件，就说明编译成功了。

 3、将slim加入PYTHONPATH

3.1 在/home/lg/anaconda3/envs/tensorflow-gpu/lib/python3.6/site-packages/tensorflow/models/research/slim文件夹下找到setup.py

在slim文件夹下执行如下命令(如下图) ：

python setup.py build

python setup.py install

 

 3.2 输入：sudo gedit ~/.bashrc

在如下文件最后添加形如：export PYTHONPATH=$PYTHONPATH:'pwd':'pwd'/slim的路径，此处'pwd'为research文件夹的路径

我的路径：

export PYTHONPATH=$PYTHONPATH:/home/lg/anaconda3/envs/tensorflow-gpu/lib/python3.6/site-packages/tensorflow/models/research:/home/lg/anaconda3/envs/tensorflow-gpu/lib/python3.6/site-packages/tensorflow/models/research/slim

输入：source  ~/.bashrc 使修改立即生效

 3.3 运行models/research下的setup.py，执行如下命令：

python setup.py build

python setup.py install

4、测试安装是否成功

在research下执行如下命令：

python object_detection/builders/model_builder_test.py

出现如下结果，说明已安装成功：

 

5、 执行已经训练好的模型

**安装tensorflow的anaconda环境，在research下输入：jupyter notebook

 

 打开object_detection文件下的object_detection_tutorial.ipynb，如下图：

 

 点击单元格，再点击运行所有

运行结果如下，成功！！！！