bdd100k数据集标签转COO再转YOLO程序

#-*-coding:utf-8-*- import sys import os import csv import xml.etree.ElementTree as Et from xml.etree.ElementTree import Element, ElementTree from PIL import Image import json from xml.etree.ElementTree import dump # Common Data format """ { "filename" : { "size" : { "width" : <string> "height" : <string> "depth" : <string> } "objects" : { "num_obj" : <int> "<index>" : { "name" : <string> "bndbox" : { "xmin" : <float> "ymin" : <float> "xmax" : <float> "ymax" : <float> } } ... } } """ # XML Data format """ { "filename" : <XML Object> ... } """ def printProgressBar (iteration, total, prefix = '', suffix = '', decimals = 1, length = 100, fill = '█'): """ Call in a loop to create terminal progress bar @params: iteration - Required : current iteration (Int) total - Required : total iterations (Int) prefix - Optional : prefix string (Str) suffix - Optional : suffix string (Str) decimals - Optional : positive number of decimals in percent complete (Int) length - Optional : character length of bar (Int) fill - Optional : bar fill character (Str) """ percent = ("{0:." + str(decimals) + "f}").format(100 * (iteration / float(total))) filledLength = int(length * iteration // total) bar = fill * filledLength + '-' * (length - filledLength) print('\r%s|%s| %s%% (%s/%s) %s' % (prefix, bar, percent, iteration, total, suffix), end = '\r') # Print New Line on Complete if iteration == total: print("\n") class VOC: """ Handler Class for VOC PASCAL Format """ def xml_indent(self, elem, level=0): i = "\n" + level * "\t" if len(elem): if not elem.text or not elem.text.strip(): elem.text = i + "\t" if not elem.tail or not elem.tail.strip(): elem.tail = i for elem in elem: self.xml_indent(elem, level + 1) if not elem.tail or not elem.tail.strip(): elem.tail = i else: if level and (not elem.tail or not elem.tail.strip()): elem.tail = i def generate(self, data): try: xml_list = {} progress_length = len(data) progress_cnt = 0 printProgressBar(0, progress_length, prefix='\nVOC Generate:'.ljust(15), suffix='Complete', length=40) for key in data: element = data[key] xml_annotation = Element("annotation") xml_size = Element("size") xml_width = Element("width") xml_width.text = element["size"]["width"] xml_size.append(xml_width) xml_height = Element("height") xml_height.text = element["size"]["height"] xml_size.append(xml_height) xml_depth = Element("depth") xml_depth.text = element["size"]["depth"] xml_size.append(xml_depth) xml_annotation.append(xml_size) xml_segmented = Element("segmented") xml_segmented.text = "0" xml_annotation.append(xml_segmented) if int(element["objects"]["num_obj"]) < 1: return False, "number of Object less than 1" for i in range(0, int(element["objects"]["num_obj"])): xml_object = Element("object") obj_name = Element("name") obj_name.text = element["objects"][str(i)]["name"] xml_object.append(obj_name) obj_pose = Element("pose") obj_pose.text = "Unspecified" xml_object.append(obj_pose) obj_truncated = Element("truncated") obj_truncated.text = "0" xml_object.append(obj_truncated) obj_difficult = Element("difficult") obj_difficult.text = "0" xml_object.append(obj_difficult) xml_bndbox = Element("bndbox") obj_xmin = Element("xmin") obj_xmin.text = element["objects"][str(i)]["bndbox"]["xmin"] xml_bndbox.append(obj_xmin) obj_ymin = Element("ymin") obj_ymin.text = element["objects"][str(i)]["bndbox"]["ymin"] xml_bndbox.append(obj_ymin) obj_xmax = Element("xmax") obj_xmax.text = element["objects"][str(i)]["bndbox"]["xmax"] xml_bndbox.append(obj_xmax) obj_ymax = Element("ymax") obj_ymax.text = element["objects"][str(i)]["bndbox"]["ymax"] xml_bndbox.append(obj_ymax) xml_object.append(xml_bndbox) xml_annotation.append(xml_object) self.xml_indent(xml_annotation) xml_list[key.split(".")[0]] = xml_annotation printProgressBar(progress_cnt + 1, progress_length, prefix='VOC Generate:'.ljust(15), suffix='Complete', length=40) progress_cnt += 1 return True, xml_list except Exception as e: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] msg = "ERROR : {}, moreInfo : {}\t{}\t{}".format(e, exc_type, fname, exc_tb.tb_lineno) return False, msg @staticmethod def save(xml_list, path): try: path = os.path.abspath(path) progress_length = len(xml_list) progress_cnt = 0 printProgressBar(0, progress_length, prefix='\nVOC Save:'.ljust(10), suffix='Complete', length=40) for key in xml_list: xml = xml_list[key] filepath = os.path.join(path, "".join([key, ".xml"])) ElementTree(xml).write(filepath) printProgressBar(progress_cnt + 1, progress_length, prefix='VOC Save:'.ljust(15), suffix='Complete', length=40) progress_cnt += 1 return True, None except Exception as e: exc_type, exc_obj, exc_tb = sys.exc_info() fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1] msg = "ERROR : {}, moreInfo : {}\t{}\t{}".format(e, exc_type, fname, exc_tb.tb_lineno) return False, msg @staticmethod def parse(path): try: (dir_path, dir_names, filenames) = next(os.walk(os.path.abspath(path))) data = {} progress_length = len(filenames) progress_cnt = 0 printProgressBar(0, progress_length, prefix='\nVOC Parsing:'.ljust(15), suffix='Complete', length=40)