使用yolo进行数据训练将xml文件转换为yolo所需的归一化txt文件时,出现以下错误:
xml转换为txt文件代码:
import glob
import xml.etree.ElementTree as ET
# 类名
class_names = ['helmet','person']
# xml文件路径,train_images只需改为val_images就可以处理val_images的了
path = r'' # xml文件路径
save_path = r'' # 归一化后txt文件保存的路径
# 转换一个xml文件为txt
def single_xml_to_txt(xml_file):
tree = ET.parse(xml_file)
root = tree.getroot()
# 保存的txt文件路径
tmp = xml_file.split('\\')[-1]
num = tmp.split('.')[0]
print("fileName = ", num