Pascal Voc格式转ICDAR2015格式的脚本

PascalVoc转ICDAR2015脚本

原创于 2020-11-23 10:52:40 发布 · 509 阅读

5 ·

CC 4.0 BY-SA版权

计算机视觉专栏收录该内容

3 篇文章

订阅专栏

本文提供了一段Python脚本，用于将PascalVoc格式的数据转换为ICDAR2015格式，适用于OCR项目中数据集格式的转换需求。脚本遍历XML文件，读取边界框坐标并将其写入文本文件。

最近做OCR的项目，需要将Pascal Voc的格式转换为ICDAR2015的格式，所以写了这个转换脚本。

import os
import xml.etree.ElementTree as ET

xmldir='Annotations'
gtdir='ICDAR_gt'

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

xmlfiles=os.listdir(xmldir)
for xmlfile in xmlfiles:
    tree = ET.parse(xmldir+'/'+xmlfile)
    root = tree.getroot()
    gt_name='gt_'+xmlfile.split('.')[0]+'.txt'
    with open(gtdir+'/'+gt_name,'a') as f:
        for obj in root.iter('object'):
            difficult = obj.find('difficult').text
            number = obj.find('name').text
            xmlbox = obj.find('bndbox')
            xmin=int(xmlbox.find('xmin').text)
            xmax=int(xmlbox.find('xmax').text)
            ymin=int(xmlbox.find('ymin').text)
            ymax=int(xmlbox.find('ymax').text)
            print('xmin:',xmin)
            print('xmax:',xmax)
            print('ymin:',ymin)
            print('ymax:',ymax)
            x1=xmin
            y1=ymin
            x2=xmax
            y2=ymin
            x3=xmax
            y3=ymax
            x4=xmin
            y4=ymax
            line=str(x1)+','+str(y1)+','+str(x2)+','+str(y2)+','+str(x3)+','+str(y3)+','+str(x4)+','+str(y4)+','+number+'\n'
            f.write(line)