使用faster rcnn训练自己的数据-制作数据集

之前配置好了faster rcnn，生成了demo，激动死了，准备来试试自己的数据，今天先制作数据集~
1.首先制作需要的数据格式，在这里我是根据这篇博客http://blog.csdn.net/sinat_30071459/article/details/50723212来制作的，根据自己的要求做了少许的修改。对图像数据进行重命名的matlab程序如下：
clc; 

clear;  

maindir='E:\faster_rcnn-master\正样本\'; 

name_long=6; %图片名字的长度，如000123.jpg为6,最多9位,可修改 

num_begin=1; %图像命名开始的数字如000123.jpg开始的话就是123 

 

subdir = dir(maindir); 

n=1; 

 

for i = 1:length(subdir) 

  if ~strcmp(subdir(i).name ,'.') && ~strcmp(subdir(i).name,'..')

     subsubdir = strcat(maindir,subdir(i).name);  

            img=imread(subsubdir); 

            str=num2str(num_begin,'%09d'); 

            newname=strcat(str,'.jpg'); 

            newname=newname(end-(name_long+3):end); 

            system(['rename ' subsubdir ' ' newname]); 

            num_begin=num_begin+1;  

  end   

end 

根据自己的要求做了修改，确定好路径后直接在原来的路径上对所有图片重命名，效果如下：



2.画方框
   因为我的系统是win10的，所以博客里的工具都用不上，不过没关系，大家可以根据自己的要求写一个画方框的小程序，由于我的数据集都是单目标的，所以只设定了一个目标名称human~代码如下：
%%%%%%以下程序为画框

clc;

clear;

maindir='E:\faster_rcnn-master\正样本\'; 

savefile='E:\faster_rcnn-master\正样本\output.txt';

sub=dir(maindir);

 fp = fopen(savefile,'wt');

for i=1:length(sub)

    if ~strcmp(sub(i).name,'.') && ~strcmp(sub(i).name,'..')

        adress=strcat(maindir,sub(i).name);

        picture=imread(adress);

         imshow(picture);

          k = waitforbuttonpress;              % 等待鼠标按下

         point1 = get(gca,'CurrentPoint');    % 鼠标按下了

         finalRect = rbbox;                   %

         point2 = get(gca,'CurrentPoint');    % 鼠标松开了

         point1 = point1(1,1:2);              % 提取出两个点

         point2 = point2(1,1:2);

         %%%将两个点另存到文件里

         string=strcat(sub(i).name,' human  ');

         fprintf(fp, '%s%s%s%s%s%s%s', string,' ',num2str(point1(1)),' ',num2str(point1(2)),' ',num2str(point2(1)),' ',num2str(point2(2)));

         fprintf(fp,'\n');

    end

end

fclose(fp);

效果跟要求的一样，如下：



分别是：图片名称 目标名 左上角坐标  右下角坐标
直接复制进去代码，修改你的图片路径（注意是全路径，最后一个有个反斜杠，不然路径会不正确），然后弹出一个图片画一个框即可，松开鼠标自动进入下一张图片的绘制，还好我的数据量不是很大。。。500多张，真的画得想吐血了。。。

3 将txt转成xml
   直接用的博客里面的程序，地址http://download.csdn.net/download/sinat_30071459/9531172，修改路径后即可，会弹出你做的标记的框框。。。会一直闪到结束，效果如下：



生成后的文件效果如下:



4  新建一个文件夹，名字为Annotations，将xml文件全部放到该文件夹里。
5 新建一个文件夹，名字为JPEGImages，将所有的训练图片放到该文件夹里
6 新建文件夹，命名为ImageSets，在ImageSets里再新建文件夹，命名为Main。
然后生成4个txt文件，主要就是打乱顺序进行测试和训练，在这里使用博客中的代码，注意我的matlab版本是r2014a，使用博客中的代码生成的文件格式有问题，没有换行，在这里只需要改一个地方，把fopen(~,'w')中的参数‘w’改成'wt'就行了。运行程序之后会自动在Main下生成4个文件。

7 数据算制作好了，然后新建一个文件夹把这三个文件放进去，就是Annotations、ImageSets和JPEGImages。
  或者替换voc2007数据集中的Annotations、ImageSets和JPEGImages。后期方便训练。

OK~制作完毕~辛苦了辛苦了~吃饭，晚上接着干~