obj = doc.createElement('object')

name = doc.createElement('name')

# name.appendChild(doc.createTextNode(datas[0]))

name.appendChild(doc.createTextNode('face'))

obj.appendChild(name)

pose = doc.createElement('pose')

pose.appendChild(doc.createTextNode('Unspecified'))

obj.appendChild(pose)

truncated = doc.createElement('truncated')

truncated.appendChild(doc.createTextNode(str(0)))

obj.appendChild(truncated)

difficult = doc.createElement('difficult')

difficult.appendChild(doc.createTextNode(str(0)))

obj.appendChild(difficult)

bndbox = doc.createElement('bndbox')

xmin = doc.createElement('xmin')

xmin.appendChild(doc.createTextNode(str(datas[1])))

bndbox.appendChild(xmin)

ymin = doc.createElement('ymin')

ymin.appendChild(doc.createTextNode(str(datas[2])))

bndbox.appendChild(ymin)

xmax = doc.createElement('xmax')

xmax.appendChild(doc.createTextNode(str(datas[3])))

bndbox.appendChild(xmax)

ymax = doc.createElement('ymax')

if '\r' == str(datas[4])[-1] or '\n' == str(datas[4])[-1]:

data = str(datas[4])[0:-1]

else:

data = str(datas[4])

ymax.appendChild(doc.createTextNode(data))

bndbox.appendChild(ymax)

obj.appendChild(bndbox)

wrong_list = []

for walk in os.walk(labels):

# os.walk遍历文件夹所有的文件

for each in walk[2]:

print(each)

fidin=open(walk[0] + '/'+ each,'r')

objIndex = 0

for data in islice(fidin, 1, None):

objIndex += 1

data=data.strip('\n')

datas = data.split(' ')

# 审核机制

# if 5 != len(datas):

# print 'bounding box information error:{}'.format(datas[0])

# # 检验一行，是否为5个元素，不是则报错

# continue

if 5 != len(datas):

wrong_list.append(each)

# 检验一行，是否为5个元素，不是则跳过

pass

pictureName = each.replace('txt', 'jpg')

imageFile = imgpath + pictureName

print(imageFile)

img = cv2.imread(imageFile)

imgSize = img.shape

if 1 == objIndex:

xmlName = each.replace('.txt', '.xml')

f = open(xmlpath_new + xmlName, "w")

doc = Document()

annotation = doc.createElement('annotation')

doc.appendChild(annotation)

folder = doc.createElement('folder')

folder.appendChild(doc.createTextNode(foldername))

annotation.appendChild(folder)

filename = doc.createElement('filename')

filename.appendChild(doc.createTextNode(pictureName))

annotation.appendChild(filename)

source = doc.createElement('source')

database = doc.createElement('database')

database.appendChild(doc.createTextNode('My Database'))

source.appendChild(database)

source_annotation = doc.createElement('annotation')

source_annotation.appendChild(doc.createTextNode(foldername))

source.appendChild(source_annotation)

image = doc.createElement('image')

image.appendChild(doc.createTextNode('flickr'))

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)

name = doc.createElement('name')

name.appendChild(doc.createTextNode('matt'))

owner.appendChild(name)

annotation.appendChild(owner)

size = doc.createElement('size')

width = doc.createElement('width')

width.appendChild(doc.createTextNode(str(imgSize[1])))

size.appendChild(width)

height = doc.createElement('height')

height.appendChild(doc.createTextNode(str(imgSize[0])))

size.appendChild(height)

depth = doc.createElement('depth')

depth.appendChild(doc.createTextNode(str(imgSize[2])))

size.appendChild(depth)

annotation.appendChild(size)

segmented = doc.createElement('segmented')

segmented.appendChild(doc.createTextNode(str(0)))

annotation.appendChild(segmented)

try:

annotation.appendChild(insertObject(doc, datas))

except:

wrong_list.append(each)

else:

try:

annotation.appendChild(insertObject(doc, datas))

except:

wrong_list.append(each)

try:

f.write(doc.toprettyxml(indent = ' '))

f.close()

fidin.close()

except:

pass