VOC数据提取自己需要的类生成XML标签

转载自:http://blog.csdn.net/samylee/article/details/61919677
公开数据集VOC里面有20类,若小伙伴需要提取特定的类别供深度学习训练的话,可以参考博主的这篇博客,如有不正,欢迎指出。
注意:
此python程序经测试VOC2007有效,其他年份的数据只需对程序稍作修改即可

import os  
import os.path  
import shutil  

fileDir_ann = "E:\\voc_year\\VOCdevkit\\VOC2007\\Annotations"  
fileDir_img = "E:\\voc_year\\VOCdevkit\\VOC2007\\JPEGImages\\"  
saveDir_img = "E:\\voc_year\\VOCdevkit\\VOC2007\\JPEGImages_ssd\\"  

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 = "E:\\voc_year\\VOCdevkit\\VOC2007\\Annotations_ssd\\"  
        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>bicycle</name>\n'  
        classes2 = '\t\t<name>motorbike</name>\n'  
        classes3 = '\t\t<name>bus</name>\n'  
        classes4 = '\t\t<name>car</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()  

但是我发现这个程序在使用过程中是存在bug的,将其修改为:

import os  
import os.path  
import shutil  

fileDir_ann = "F:\\VOC_mine\\Annotations"  
fileDir_img = "F:\\VOC_mine\\JPEGImages\\"  
saveDir_img = "F:\\VOC_mine\\JPEGImages_ssd\\"  

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 = "F:\\VOC_mine\\Annotations_ssd\\"  
        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[:]

        classes1 = '\t\t<name>bicycle</name>\n'  
        classes2 = '\t\t<name>motorbike</name>\n'  
        classes3 = '\t\t<name>bus</name>\n'  
        classes4 = '\t\t<name>car</name>\n'

        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)):
            names['block%d'%k] = []
            for j in range(0,len(classes)):  
                if classes[j] in lines[ind_start[i]+1]:  
                    a = ind_start[i]
                    for o in range(ind_end[i]-ind_start[i]+1):
                        names['block%d'%k].append(lines[a+o])
                    break  
            i += 1  

        string_start = lines[0:ind_start[0]]           
        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]  

        string_start += lines[ind_end[-1]+1:]    
        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()  

这样可以应用于各个年份的VOC数据!

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在VS Code中将XML格式转换为VOC格式,您可以按照以下步骤进行操作: 1. 首先,确保您已经安装了Python和所需的库。您可以使用以下命令安装所需的库: ``` pip install xmltodict ``` 2. 创建一个Python脚本文件(例如convert_xml_to_voc.py),并在VS Code中打开它。 3. 在脚本文件中,导入所需的库和模块: ```python import os import xml.etree.ElementTree as ET import xmltodict ``` 4. 定义一个函数来解析XML文件并生成VOC格式的标注数据: ```python def convert_xml_to_voc(xml_file): with open(xml_file, 'r') as f: xml_data = f.read() data_dict = xmltodict.parse(xml_data) # 提取图像尺寸信息 image_width = int(data_dict['annotation']['size']['width']) image_height = int(data_dict['annotation']['size']['height']) image_depth = int(data_dict['annotation']['size']['depth']) # 创建VOC格式的标注数据字符串 voc_data = f"{image_width}\n{image_height}\n{image_depth}\n" # 提取目标对象信息并生成VOC格式的标注数据字符串 for obj in data_dict['annotation']['object']: xmin = int(obj['bndbox']['xmin']) ymin = int(obj['bndbox']['ymin']) xmax = int(obj['bndbox']['xmax']) ymax = int(obj['bndbox']['ymax']) label = obj['name'] voc_data += f"{label} {xmin} {ymin} {xmax} {ymax}\n" return voc_data ``` 5. 定义一个函数来遍历指定目录下的所有XML文件,并调用上面的函数进行转换: ```python def batch_convert_xml_to_voc(xml_dir, output_dir): for xml_file in os.listdir(xml_dir): if xml_file.endswith('.xml'): xml_path = os.path.join(xml_dir, xml_file) voc_data = convert_xml_to_voc(xml_path) # 生成VOC格式的标注数据文件 output_file = os.path.splitext(xml_file)[0] + '.txt' output_path = os.path.join(output_dir, output_file) with open(output_path, 'w') as f: f.write(voc_data) ``` 6. 在脚本文件中,调用上面的函数并传入您的XML文件目录和输出目录: ```python xml_dir = 'path/to/xml/files' output_dir = 'path/to/output/directory' batch_convert_xml_to_voc(xml_dir, output_dir) ``` 7. 保存并运行脚本文件,它将遍历指定目录下的所有XML文件,并将转换后的VOC格式标注数据保存到指定的输出目录中。 请注意,上述代码仅提供了一个简单的示例,并假设XML文件的结构符合您的需求。您可能需要根据自己的数据集结构进行适当的修改。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值