YOLO训练数据集制作-voc格式 xml 生成txt 脚本_天天放羊的博客_voc xml生成txt

1. 生成train文件

import os
import os.path
 
path = "data/putao2/images/"
 
 
for filenames in os.walk(path):
    filenames = list(filenames)
    filenames = filenames[2]
    for filename in filenames:
        print(filename)
        with open ("data/putao2/train.txt",'a') as f:
            f.write(path+filename+'\n')
import os
import os.p