【tensorflow：目标检测】 Windows 下 配置安装 Faster RCNN

目录

 

最好使用python3.5

参考链接：

1.、确保TensorFlow和相关的依赖安装成功

2、下载faster rcnn程序包，

3、在tensorflow这个虚拟环境中安装python依赖库

4、配置

5、下载VOC2007格式的数据集

6、下载预训练的VGG16网络模型：

7、现在就可以训练模型了：

---

最好使用python3.5

不要使用3.6，要不然容易出错

参考链接：

https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5

主要是对这个英文的翻译

最好配置anaconda，同时创建一个tensorflow环境，使用GPU版本的tensorflow。

 如图所示我在anaconda中创建了一个名为tensorflow的虚拟环境，然后在里面安装了tensorflow-GPU。同时我还把这个虚拟python环境添加了环境变量。

有一个虚拟环境的好处在于，无论你如何安装包，都不会影响到anaconda的根python。

如上图，D:\Anaconda3\Scripts 表示base python

D:\Anaconda3\envs\tensorflow\Scripts  表示tensorflow使用的 python

1.、确保TensorFlow和相关的依赖安装成功

2、下载faster rcnn程序包，

这是我的GitHub，已经fork 。

：https://github.com/ChaoflyLi/Faster-RCNN-TensorFlow-Python3.5

英语好的可以看一下里面的readme，不好也没关系，跟着一步一步来就可以了

3、在tensorflow这个虚拟环境中安装python依赖库

activate tensorflow
 
pip install cython
 
pip install python-opencv
 
pip install easydict

最好使用离线安装，如何离线安装请参考我的博客：https://blog.csdn.net/ChaoFeiLi/article/details/86606128   在3.3中有介绍

4、配置

在虚拟tensorflow的环境下，cmd中使用cd命令到解压后的文件夹下../data/coco/PythonAPI目录下

执行以下代码 

python setup.py build_ext --inplace
 
python setup.py build_ext install

转到./lib/utils并运行

python setup.py build_ext --inplace

我参考的博客中 https://blog.csdn.net/tuoyakan9097/article/details/81782257，没有提到这个，着实坑了我一把，在第7不一直报错，真的是服了。

若是报错：

running build_ext
building 'pycocotools._mask' extension
error: Unable to find vcvarsall.bat

安装visual studio 2015，安装时默认的语言是C#和VB，需要把C++也勾上。

如果已经安装了VS 2015，检查VS安装目录下是否有vcvarsall.bat文件，例如我的路径为：C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC，如果没有此文件，表明安装时是默认安装的，还缺少C++编译文件，这个时候可以打开VS，新建项目，语言选择C++，此时VS会让你安装一个工具，安装后就好了。

请参考下列了解，安装VS 2015中的C++
参考链接：https://blog.csdn.net/childbor/article/details/81286534

然后就好了

5、下载VOC2007格式的数据集

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

也可以从网盘下载

链接：https://pan.baidu.com/s/18FIHuJ_GaY7K4xR8cqw_DQ 
提取码：rujr 
复制这段内容后打开百度网盘手机App，操作更方便哦

将下载后的三个压缩包解压到同一个文件夹下，同时选中三个压缩包，右键，然后选择解压到当前文件夹

可以得到VOCDevkit文件夹，将VOCDevkit重命名为VOCDevkit2007，然后将这个文件夹拷贝到你自己的Faster-RCNN中data目录下。

 

6、下载预训练的VGG16网络模型：

下载地址：http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz

下载之后解压，文件重命名为vgg16.ckpt，新建文件夹imagenet_weights，把vgg16.ckpt放到imagenet_weights下，再将imagenet_weights文件夹拷贝到data文件夹下，

这样，VGG16网络模型的地址应该类似于这样：../data/imagenet_weights/vgg16.ckpt

我们也可以用别的模型，模型下载链接：https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models

7、现在就可以训练模型了：

在faster rcnn目录下运行命令：

python train.py

OK，搞定，这时候，就是在用VOCDevkit2007下的示例图片训练分类模型了。第一次运行时可能会报错说找不到matplotlib，scipy，PIL（对应的是pillow库）等，安装这些库就好了