# -*- coding: utf-8 -*-import xml.etree.ElementTree as ET
import os
from os import getcwd
sets =['train_all','test_all']
classes =['CL','CE1','CE2','CE3','CE4','CE5','AE1','AE2','AE3']# 改成自己的类别
abs_path = os.getcwd()print(abs_path)defconvert(size, box):
dw =1./(size[0])
dh =1./(size[1])
x =(box[0]+ box[1])/2.0-1
y =(box[2]+ box[3])/2.0-1
w = box[1]- box[0]
h = box[3]- box[2]
x = x * dw
w = w * dw
y = y * dh
h = h * dh
return x, y, w, h
defconvert_annotation(image_id):
in_file =open('/Users/h1y/Documents/mac2server/echi/echi/Annotations/%s.xml'%(image_id), encoding='UTF-8')
out_file =open('/Users/h1y/Documents/mac2server/echi/echi/labels/%s.txt'%(image_id),'w')
tree = ET.parse(in_file)
root = tree.getroot()
size = root.find('size')
w =int(size.find('width').text)
h =int(size.find('height').text)for obj in root.iter('object'):# difficult = obj.find('difficult').text# difficult = obj.find('Difficult').text
cls = obj.find('name').text
# if cls not in classes or int(difficult) == 1:# continueif cls notin classes:continue
cls_id = classes.index(cls)
xmlbox = obj.find('bndbox')
b =(float(xmlbox.find('xmin').text),float(xmlbox.find('xmax').text),float(xmlbox.find('ymin').text),float(xmlbox.find('ymax').text))
b1, b2, b3, b4 = b
# 标注越界修正if b2 > w:
b2 = w
if b4 > h:
b4 = h
b =(b1, b2, b3, b4)
bb = convert((w, h), b)
out_file.write(str(cls_id)+" "+" ".join([str(a)for a in bb])+'\n')
wd = getcwd()for image_set in sets:ifnot os.path.exists('/Users/h1y/Documents/mac2server/echi/echi/labels/'):
os.makedirs('/Users/h1y/Documents/mac2server/echi/echi/labels/')
image_ids =open('/Users/h1y/Documents/mac2server/echi/echi/ImageSets/Main/%s.txt'%(image_set)).read().strip().split()
list_file =open('/Users/h1y/Documents/mac2server/echi/echi/%s.txt'%(image_set),'w')for image_id in image_ids:
list_file.write(abs_path +'/images/train/%s.jpg\n'%(image_id))
convert_annotation(image_id)
list_file.close()