''' author ： Hatcher W Email： hcwang@semi.ac.cn use: 用于创建存储目标检测结果的xml文件 input:网络运算输出结果 output：xml文件 ''' import numpy as np from lxml import etree from common import global_para as gp import xmltodict import json,time class GEN_Annotations: def __init__(self, filename): self.root = etree.Element("annotation") child1 = etree.SubElement(self.root, "folder") child1.text = "Hatcher" child2 = etree.SubElement(self.root, "filename") child2.text = filename child3 = etree.SubElement(self.root, "source") child4 = etree.SubElement(child3, "annotation") child4.text = "Bird_detection" child5 = etree.SubElement(child3, "database") child5.text = "Unknown" child6 = etree.SubElement(child3, "image") child6.text = "flickr" child7 = etree.SubElement(child3, "flickrid") child7.text = "35435" def set_size(self, witdh, height, channel): size = etree.SubElement(self.root, "size") widthn = etree.SubElement(size, "width") widthn.text = str(witdh) heightn = etree.SubElement(size, "height") heightn.text = str(height) channeln = etree.SubElement(size, "depth") channeln.text = str(channel) def savefile(self, filename): tree = etree.ElementTree(self.root) tree.write(filename, pretty_print=True, xml_declaration=False, encoding='utf-8') def add_pic_attr(self, label, xmin, ymin, xmax, ymax,score): object = etree.SubElement(self.root, "object") namen = etree.SubElement(object, "name") namen.text = str(label).replace("tensor(","").replace("., device='cuda:0')","") bndbox = etree.SubElement(object, "bndbox") xminn = etree.SubElement(bndbox, "xmin") xminn.text = str(xmin) yminn = etree.SubElement(bndbox, "ymin") yminn.text = str(ymin) xmaxn = etree.SubElement(bndbox, "xmax") xmaxn.text = str(xmax) ymaxn = etree.SubElement(bndbox, "ymax") ymaxn.text = str(ymax) scoren = etree.SubElement(bndbox, "score") scoren.text = str(score) def add_pic_attr_bird(self, label, xmin, ymin, xmax, ymax): object = etree.SubElement(self.root, "object") namen = etree.SubElement(object, "name") namen.text = label bndbox = etree.SubElement(object, "bndbox") xminn = etree.SubElement(bndbox, "xmin") xminn.text = str(xmin) yminn = etree.SubElement(bndbox, "ymin") yminn.text = str(ymin) xmaxn = etree.SubElement(bndbox, "xmax") xmaxn.text = str(xmax) ymaxn = etree.SubElement(bndbox, "ymax") ymaxn.text = str(ymax) def add_pic_density(self, label, density_h): object = etree.SubElement(self.root, "object") count = etree.SubElement(object, "count") for i in density_h: density = etree.SubElement(object, "density_h") num=0 for j in i: density.attrib["att{}".format(num)] = j num+=1 class GEN_Annotations_scan(GEN_Annotations): def __init__(self, record_time,serial_number): super(GEN_Annotations,self).__init__() self.root = etree.Element("annotation") child1 = etree.SubElement(self.root, "record_time") child1.text = record_time child2 = etree.SubElement(self.root, "serial_number") child2.text = serial_number def add_pic_attr_scan(self, label, xmin, ymin, xmax, ymax): object = etree.SubElement(self.root, "data") namen = etree.SubElement(object, "species_name") namen.text = str(label) bndbox = etree.SubElement(object, "bndbox") pos_x = etree.SubElement(bndbox, "site_x") pos_x.text = str("1") pos_y = etree.SubElement(bndbox, "site_y") pos_y.text = str("1") xminn = etree.SubElement(bndbox, "xmin") xminn.text = str(xmin) yminn = etree.SubElement(bndbox, "ymin") yminn.text = str(ymin) xmaxn = etree.SubElement(bndbox, "xmax") xmaxn.text = str(xmax) ymaxn = etree.SubElement(bndbox, "ymax") ymaxn.text = str(ymax) scoren = etree.SubElement(object, "habitat") scoren.text = str("") def make_xml_scan(record_time,serial_number, boxes,labels,savename): anno = GEN_Annotations_scan(record_time,serial_number) #anno.set_size(size_y, size_x, 3) for i in range(len(boxes)): xmin = int(boxes[i][0]) ymin = int(boxes[i][1]) xmax = int(boxes[i][2]) ymax = int(boxes[i][3]) # size_x1 = size_x[i] # size_y1 = size_y[i] anno.add_pic_attr_scan(str(labels[i]), xmin, ymin, xmax, ymax) anno.savefile("./temp.xml") with open("./temp.xml", 'r', encoding='UTF-8')as json_file: xml_data = json_file.read() json_file.close() jsonstr = xmltodict.parse(xml_data, encoding='UTF-8') json_conversion = json.dumps(jsonstr["annotation"]) # 将python对象编码成Json字符串 with open(savename, 'w') as json_file: json_file.write(json_conversion) # 写入文件 json_file.close() return jsonstr["annotation"] def make_xml(filename,size_x,size_y,boxes,labels,scores,savename): anno = GEN_Annotations(filename) anno.set_size(size_y, size_x, 3) for i in range(len(boxes)): xmin = int(boxes[i][0]) ymin = int(boxes[i][1]) xmax = int(boxes[i][2]) ymax = int(boxes[i][3]) score = int(scores[i]*1000) anno.add_pic_attr(gp.BHG_chineseName[int(labels[i])], xmin, ymin, xmax, ymax,score) print("细粒度识别结果为：",gp.BHG_chineseName[int(labels[i])]) anno.savefile(savename) def make_xml_iccv(filename,size_x,size_y,boxes,savename): anno = GEN_Annotations(filename) anno.set_size(size_y, size_x, 3) for i in range(len(boxes)): xmin = int(boxes[i][0]) ymin = int(boxes[i][1]) xmax = int(boxes[i][2]) ymax = int(boxes[i][3]) anno.add_pic_attr_bird("bird", xmin, ymin, xmax, ymax) anno.savefile(savename) if __name__ == '__main__': filename = "000001.jpg" anno = GEN_Annotations(filename) anno.set_size(1280, 720, 3) density = np.array([4,4],dtype=np.float64) density = [[0.123,93.132213,0.2132131,0.32132131],[0.123,93.132213,0.2132131,0.32132131],[0.123,93.132213,0.2132131,0.32132131],[0.123,93.132213,0.2132131,0.32132131]] anno.add_pic_density("mouse", density) anno.savefile("00001.xml")