将xml格式的标注文件转为csv格式
方便后续转为tfrecord格式供tensorflow使用
注意,该代码仅使用矩形框标注方法
import os
import glob
import pandas as pd
import xml.etree.ElementTree as ET
path = 'D:/Temp/4856/all_class/test_xml'#xml标注文件路径
save_path = ( path + '/')#csv文件保存路径
def xml_to_csv(path):
xml_list = []
for xml_file in glob.glob(path + '/*.xml'):
tree = ET.parse(xml_file)
root = tree.getroot()