yolo的txt转xml

from xml.dom.minidom import Document
import os
import os.path
from PIL import Image
ann_path=‘D:\E\txt\’#txt 文件夹
img_path=‘D:\E\img\’#图片文件夹
xml_path=‘D:\E\Annotations\’#xml 文件存放地址

def CreateXml():

import xml.dom.minidom

impl = xml.dom.minidom.getDOMImplementation()

dom = impl.createDocument(None, ‘employees’ , None)

root = dom.documentElement

employee = dom.createElement( ‘employee’ )

root.appendChild(employee)

nameE = dom.createElement( ‘name’ )

nameT = dom.createTextNode( ‘linux’ )

nameE.appendChild(nameT)

employee.appendChild(nameE)

ageE = dom.createElement( ‘age’ )

ageT = dom.createTextNode( ‘30’ )

ageE.appendChild(ageT)

employee.appendChild(ageE)

f = open( ‘employees2.xml’ , ‘w’)

dom.writexml(f, addindent = ’ ’ , newl = ‘\n’ ,encoding = ‘utf-8’ )

f.close()

CreateXml()

if not os.path.exists(xml_path):
os.mkdir(xml_path)
classes=[‘1’]
def writeXml(tmp,imgname,w,h,objbud,wxml):
doc=Document()

annotation=doc.createElement('annotation')
doc.appendChild(annotation)
folder=doc.createElement('folder')
annotation.appendChild(folder)
folder_txt=doc.createTextNode('VOC2019')
folder.appendChild(folder_txt)
filename=doc.createElement('filename')
annotation.appendChild(filename)
filename_txt=doc.createTextNode(imgname)
filename.appendChild(filename_txt)
source=doc.createElement('source')
annotation.appendChild(source)
database=doc.createElement('database')
source.appendChild(database)
database_txt=doc.createTextNode('The VOC2019 Database')
database.appendChild(database_txt)
annotation_new=doc.createElement('annotation')
source.appendChild(annotation_new)
annotation_new_txt=doc.createTextNode('PASCAL VOC2019')
annotation_new.appendChild(annotation_new_txt)
image=doc.createElement('image')
source.appendChild(image)
image_txt=doc.createTextNode('flickr')
image.appendChild(image_txt)
size=doc.createElement('size')
annotation.appendChild(size)
width=doc.createElement('width')
size.appendChild(width)
width_txt=doc.createTextNode(str(w))
width.appendChild(width_txt)
height=doc.createElement('height')
size.appendChild(height)
height_txt=doc.createTextNode(str(h))
height.appendChild(height_txt)
depth=doc.createElement('depth')
size.appendChild(depth)
depth_txt=doc.createTextNode('3')
depth.appendChild(database_txt)

segmented=doc.createElement('segmented')
annotation.appendChild(segmented)
segmented_txt=doc.createTextNode('0')
segmented.appendChild(segmented_txt)
for i in range(0,int(len(objbud))):
    objbuds=objbud[i].split(' ')
    print(objbuds)
    object_new=doc.createElement('object')
    annotation.appendChild(object_new)
    name=doc.createElement('name')
    object_new.appendChild(name)
    name_txt=doc.createTextNode(classes[int(objbuds[0])])
    name.appendChild(name_txt)
    pose=doc.createElement('pose')
    object_new.appendChild(pose)
    pose_txt=doc.createTextNode('Unspecified')
    pose.appendChild(pose_txt)
    truncated=doc.createElement('truncated')
    object_new.appendChild(truncated)
    truncated_txt=doc.createTextNode('0')
    truncated.appendChild(truncated_txt)
    difficult=doc.createElement('difficult')
    difficult_txt=doc.createTextNode('0')
    difficult.appendChild(difficult_txt)
    bndbox=doc.createElement('bndbox')
    object_new.appendChild(bndbox)

    xmin=doc.createElement('xmin')
    bndbox.appendChild(xmin)
    xmin_txt=doc.createTextNode(str(int((float(objbuds[1])*w-float(objbuds[3])*w/2.0))))
    xmin.appendChild(xmin_txt)

    ymin=doc.createElement('ymin')
    bndbox.appendChild(ymin)
    ymin_txt=doc.createTextNode(str(int((float(objbuds[2])*h-float(objbuds[4])*h/2.0))))
    ymin.appendChild(ymin_txt)

    xmax=doc.createElement('xmax')
    bndbox.appendChild(xmax)
    xmax_txt=doc.createTextNode(str(int((float(objbuds[1])*w+float(objbuds[3])*w/2.0))))
    xmax.appendChild(xmax_txt)

    ymax=doc.createElement('ymax')
    bndbox.appendChild(ymax)
    ymax_txt=doc.createTextNode(str(int((float(objbuds[2])*h+float(objbuds[4])*h/2.0))))
    ymax.appendChild(ymax_txt)
tempfile=tmp+'\\'+'test.xml'
with open(tempfile,'w') as f:
    f.write(doc.toprettyxml(indent='\t'))
rewrite=open(tempfile,'r')
lines=rewrite.read().split('\n')
newliens=lines[1:len(lines)-1]
fw=open(wxml,'w')
for i in range(0,len(newliens)):
    fw.write(newliens[i]+'\n')
fw.close()
rewrite.close()

os.remove(tempfile)

return

for files in os.walk(ann_path):
temp = ‘D:\E\temp’
if not os.path.exists(temp):
os.mkdir(temp)
for file in files[2]:
img_name = os.path.splitext(file)[0] + ‘.jpg’
fileimgpath = img_path + img_name
im = Image.open(fileimgpath)
width = int(im.size[0])
height = int(im.size[1])
filelabel = open(ann_path + file, ‘r’)
lines = filelabel.read().split(’\n’)
obj = lines[:len(lines) - 1]
#print(obj)
filename = xml_path + os.path.splitext(file)[0] + ‘.xml’
writeXml(temp, img_name, width, height, obj, filename)

os.rmdir(temp)