实用小程序(1)之xml转txt
这段时间在看学习yolo系列算法,在训练自己的训练集时,需要读取标注的xml文件的类别并保存为txt文本格式,在网上找了很久也没有发现相关的代码,所以就自己写了一个需要的同学自取啦。
(1)获取训练图片的前缀名并保存为txt
import os
import random
xmlfilepath = 'D:\\project\\windows_yolov3\\train_img\\train_image' # 这里写自己图片保存的路径
txtsavepath = 'D:\\project\\windows_yolov3\\train_img\\name.txt' #生成的txt文本保存路径
total_xml = os.listdir(xmlfilepath)
num = len(total_xml)
list = range(num)
ftrainval = open