前言
之前写过一片博客是在Windows下配置faster_rcnn demo环境 ,如果在Windows上训练数据集会出现很多问题,于是才不得不转入Ubuntu系统,现在发觉Ubuntu系统越来越好使。
1.配置caffe环境
- 首先在Ubuntu系统下配置caffe,可以参考这篇博客ubuntu16.04+anaconda2+gpu+caffe安装
- 如果按照上述博客配置好了caffe的话,先不要高兴太早,因为你电脑上的cudnn版本很可能与faster_rcnn要求不一致,所以得重新配置caffe,但是上一步的步骤还是要进行的。
- 下载faster_rcnn代码:链接: https://pan.baidu.com/s/1VQ7pGNB_pz9v12APdQ0QKA 提取码: fi7p
下载后解压得到caffe-fast-rcnn文件夹
参考这篇博客解决cudnn版本不一致问题中的第3步骤,就是将所有cudnn版本替换,然后重新配置caffe就行。最后不要忘了在变量环境中将原先版本的caffe替换为caffe-fast-rcnn版本路径
2.配置demo
-
如果你的caffe编译好的话,这一步就很简单了,先下载模型。
cd py-faster-rcnn/data/scripts
找到fetch_faster_rcnn_models.sh文件打开里面有模型下载链接,不要使用Ubuntu系统命令下载,因为根本下载不下来,建议在Windows上使用迅雷下载,但是好像需要会员才能下载下来,于是我这提供百度云链接:链接: https://pan.baidu.com/s/1ItMcQq-3B2RuPhQaXRvYsw 提取码: irya
下载下来后复制到data文件夹下即可。
2. 进入tools文件夹,打开终端,输入命令python demo.py
如果出现结果则表示demo配置成功
3.训练数据集
- 下载voc数据集:
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
下载下来后解压成一个文件夹重命名为VOCdevkit2007,放在data文件夹下,如下图所示
如果你要是训练自己的数据集,只需按照voc数据集格式制作就行,在这里就不在叙述了。 - 下载imagenet_models模型,在py-faster-rcnn/data/scripts文件夹下找到fetch_imagenet_models.sh文件,同理在Windows下下载,这里依然提供百度云链接: https://pan.baidu.com/s/1yCMiOLZ0AHBRThPB-UUzJg 提取码: g2mw
下载下来后解压到data文件夹下即可
3.如果你要是训练自己的数据集参考这篇博客修改网络对应的参数训练自己的数据
4. 开始训练
不要参考第3步博客的训练命令会报错
cd py-faster-rcnn
./experiments/scripts/faster_rcnn_end2end.sh 0 VGG_CNN_M_1024 pascal_voc
换一种训练命令
chmod u+x /home/csm/py-faster-rcnn/tools/train_faster_rcnn_alt_opt.py
一定要加入chmod u+x命令否则会报错没有访问权限,然后输入命令
./tools/train_faster_rcnn_alt_opt.py --gpu 0 --net_name ZF --weights data/imagenet_models/ZF.v2.caffemodel --imdb voc_2007_trainval --cfg experiments/cfgs/faster_rcnn_alt_opt.yml
4.测试数据
如果上述训练数据正确的话,会在output文件夹中保存训练模型
输入命令
chmod u+x tools/test_net.py
./tools/test_net.py --gpu 0 --def models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt --net output/faster_rcnn_alt_opt/voc_2007_trainval/ZF_faster_rcnn_final.caffemodel--imdb voc_2007_test --cfg experiments/cfgs/faster_rcnn_alt_opt.yml