TXT文件转Pascal voc数据集XML格式标注文件

TXT的文件格式为 wmin，hmin，wmax，hmax

最终转换出来的效果为

代码：
from xml.dom import minidom
import cv2

txt_dirtory=r'F:\9.txt'
jpg_dirtory=r'F:\9.jpg'
img_name=jpg_dirtory.split('\\')[-1]
floder=jpg_dirtory.split('\\')[-2]
im = cv2.imread(jpg_dirtory)
w = im.shape[1]
h = im.shape[0]
d = im.shape[2]
# print w,h,d
doc = minidom.Document()   #创建DOM树对象

annotation = doc.createElement('annotation')   #创建子节点
doc.appendChild(annotation)                    #annotation作为doc树的子节点

folder = doc.createElement('folder')            
folder.appendChild(doc.createTextNode(floder))  #文本节点作为floder的子节点
annotation.appendChild(folder)                 #folder作为annotation的子节点

filename = doc.createElement('filename')
filename.appendChild(doc.createTextNode(img_name))
annotation.appendChild(filename)

filename = doc.createElement('path')
filename.appendChild(doc.createTextNode(jpg_dirtory))
annotation.appendChild(filename)

source = doc.createElement('source')
database = doc.createElement('database')
database.appendChild(doc.createTextNode("Unknown"))
source.appendChild(database)
# annotation2 = doc.createElement('annotation')
# annotation2.appendChild(doc.createTextNode("ICDAR POD2017"))
# source.appendChild(annotation2)
# image = doc.createElement('image')
# image.appendChild(doc.createTextNode("image"))
# source.appendChild(image)
# flickrid = doc.createElement('flickrid')
# flickrid.appendChild(doc.createTextNode("NULL"))
# source.appendChild(flickrid)
annotation.appendChild(source)

# owner = doc.createElement('owner')
# flickrid = doc.createElement('flickrid')
# flickrid.appendChild(doc.createTextNode("NULL"))
# owner.appendChild(flickrid)
# na = doc.createElement('name')
# na.appendChild(doc.createTextNode("cxm"))
# owner.appendChild(na)
# annotation.appendChild(owner)

size = doc.createElement('size')
width = doc.createElement('width')
width.appendChild(doc.createTextNode("%d" % w))
size.appendChild(width)
height = doc.createElement('height')
height.appendChild(doc.createTextNode("%d" % h))
size.appendChild(height)
depth = doc.createElement('depth')
depth.appendChild(doc.createTextNode("%d" % d))
size.appendChild(depth)
annotation.appendChild(size)

segmented = doc.createElement('segmented')
segmented.appendChild(doc.createTextNode("0"))
annotation.appendChild(segmented)

txtLabel = open(txt_dirtory, 'r')
boxes = [txtLabel.readline()]
for box in boxes:
    # print(box)
    box = box.split(' ')
    # print(box)
    object = doc.createElement('object')
    nm = doc.createElement('name')
    nm.appendChild(doc.createTextNode('0'))
    object.appendChild(nm)
    pose = doc.createElement('pose')
    pose.appendChild(doc.createTextNode("Unspecified"))
    object.appendChild(pose)
    truncated = doc.createElement('truncated')
    truncated.appendChild(doc.createTextNode("1"))
    object.appendChild(truncated)
    difficult = doc.createElement('difficult')
    difficult.appendChild(doc.createTextNode("0"))
    object.appendChild(difficult)
    bndbox = doc.createElement('bndbox')
    xmin = doc.createElement('xmin')
    xmin.appendChild(doc.createTextNode(box[0]))
    bndbox.appendChild(xmin)
    ymin = doc.createElement('ymin')
    ymin.appendChild(doc.createTextNode(box[1]))
    bndbox.appendChild(ymin)
    xmax = doc.createElement('xmax')
    xmax.appendChild(doc.createTextNode(box[2]))
    bndbox.appendChild(xmax)
    ymax = doc.createElement('ymax')
    ymax.appendChild(doc.createTextNode(box[3]))
    bndbox.appendChild(ymax)
    object.appendChild(bndbox)
    annotation.appendChild(object)
    savefile = open(r'F:\9.xml', 'w')
    savefile.write(doc.toprettyxml())
    savefile.close()
参考内容：
txt版本的标签转换成PASCAL VOC2007 格式的XML标签
 Python minidom模块(DOM写入和解析XML)
TXT文件转Pascal voc数据集XML格式标注文件

猜你喜欢
