Faster-rcnn训练自己的数据库
下载fasterrcnn https://github.com/ShaoqingRen/faster_rcnn
建议根据github上讲的下个训练好的模型和测试图片跑一下demo。
一、准备工作:
1、网络定义文件、预训练模型和Caffe下载
①运行fetch_data/fetch_model_ZF.m下载ImageNet-pre-trainedZF模型,用于rpn网络和fastrcnn网络模型的定义以及初始化。如此得到models文件夹。
②运行fetch_data/fetch_caffe_mex_windows_vs2013_cuda65.m下载编译好的Caffe,得到external文件夹。
也可直接打开.m文件中的链接下载。VGG模型也可同样获得。
2、数据库下载
VOC数据库下载 http://host.robots.ox.ac.uk/pascal/VOC/
训练/验证集:
测试集
注意: development kitcode and documentation必须下载(解压得到VOCdevkit文件夹),训练自己的数据库要用。
datasets文件夹结构需安排如下&#x