目录
3、在tensorflow这个虚拟环境中安装python依赖库
最好使用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库)等,安装这些库就好了