voc2007_extract_car_label

voc2007_extract_carlabel

代码参考：voc数据提取自己的类
作者代码是windows的提取了四类，我当前只需要car和bus两类，最后把bus类别全部转化为car
python代码如下：

import os  
import os.path  
import shutil  

fileDir_ann = "/home/mahnx0/data/voc2007/Annotations/"  
fileDir_img = "/home/mahnx0/data/voc2007/JPEGImages/"  
saveDir_img = "/home/mahnx0/data/voc_extract_car/img_file/"  

if not os.path.exists(saveDir_img):  
    os.mkdir(saveDir_img)  

names = locals()  

for files in os.walk(fileDir_ann):  
    for file in files[2]:  

        print file + "-->start!"  

        saveDir_ann = "/home/mahnx0/data/voc_extract_car/xml_file/"  
        if not os.path.exists(saveDir_ann):  
            os.mkdir(saveDir_ann)  

        fp = open(fileDir_ann + '/' + file)          
        saveDir_ann = saveDir_ann + file  
        fp_w = open(saveDir_ann, 'w')  
        classes = ['aeroplane','bicycle','bird','boat','bottle','bus','car','>cat<','chair','cow','diningtable','dog','horse','motorbike','pottedplant','sheep','sofa','train','tvmonitor','person']  

        lines = fp.readlines()  

        ind_start = []  
        ind_end = []  
        lines_id_start = lines[:]  
        lines_id_end = lines[:]  

        while "\t<object>\n" in lines_id_start:  
            a = lines_id_start.index("\t<object>\n")  
            ind_start.append(a)  
            lines_id_start[a] = "delete"  

        while "\t</object>\n" in lines_id_end:  
            b = lines_id_end.index("\t</object>\n")  
            ind_end.append(b)  
            lines_id_end[b] = "delete"  

        i = 0  
        for k in range(0,len(ind_start)):  
            for j in range(0,len(classes)):  
                if classes[j] in lines[ind_start[i]+1]:  
                    a = ind_start[i]  
                    names['block%d'%k] = [lines[a],lines[a+1],lines[a+2],lines[a+3],lines[a+4],lines[a+5],lines[a+6],lines[a+7],lines[a+8],lines[a+9],lines[a+10],lines[ind_end[i]]]  
                    break  
            i += 1  

        classes1 = '\t\t<name>car</name>\n'  
        classes2 = '\t\t<name>bus</name>\n'  

        string_start = lines[0:ind_start[0]]  
        string_end = [lines[len(lines)-1]]  

        a = 0  
        for k in range(0,len(ind_start)):  
            if classes1 in names['block%d'%k]:  
                a += 1  
                string_start += names['block%d'%k]  
            if classes2 in names['block%d'%k]:  
                a += 1  
                string_start += names['block%d'%k]  
 #           if classes3 in names['block%d'%k]:  
 #               a += 1  
 #               string_start += names['block%d'%k]  
 #           if classes4 in names['block%d'%k]:  
 #               a += 1  
 #               string_start += names['block%d'%k]  

        string_start += string_end   
        for c in range(0,len(string_start)):  
            fp_w.write(string_start[c])  
        fp_w.close()  

        if a == 0:  
            os.remove(saveDir_ann)  
        else:  
            name_img = fileDir_img + os.path.splitext(file)[0] + ".jpg"  
            shutil.copy(name_img,saveDir_img)  
        fp.close()  

提取完成之后文件夹如下：

把label文件中的bus批量替换成car

cd xml_file
sed -i "s/bus/car/g" `grep bus -rl ./`

制作测试集：
在xml_file下新建valimg_file, valxml_file,用sh命令xargs把后缀9的xml文件重定向，用mv命令移到valxml_file下面,把valxml_file 下xml对应的image移到valimg_file

//val xml制作
find ./ -name "*9.xml" |xargs -I{} mv {} valxml_file/
//val image制作
cd img_file
cat img_file/*9.jpg | xargs -I{} mv {} valimg_file/

至此 voc数据集提取car和bus并全部合并为car完成,为了文件名匹配
把所有文件名改成如下了：

下一步：把当前提取出来的voc格式的数据转化为kitti格式