提取训练图片并打上标签生成txt文件

最新推荐文章于 2024-06-10 01:18:13 发布

原创最新推荐文章于 2024-06-10 01:18:13 发布 · 2.9k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python #os

本示例展示了一个简单的Python脚本，用于遍历指定目录下的文件，并根据文件名中的特定模式来修改文件路径及添加标签。通过查找包含特定格式的文件名，脚本能够识别并替换相应的编号部分，最终将更新后的文件路径和对应的标签写入到一个输出文件中。

file_name=[]
file_name = os.listdir("D:\\Work\\process\\test\\polyu-all-train\\")
with open("output-train.txt", "w+") as fp:
    for i in file_name:
        for j in range(26):
            name = "_p" + str(j+1) + "_"
            if i.find(name) != -1:
                namelabel = j+1
                newname ="polyu\\"+ "p"+str(namelabel) +"\\"+i + " " + str(namelabel) +'\n'
                fp.write(newname)

结果：

polyu\p10\s10_p10_c1.jpg 10
polyu\p10\s10_p10_c2.jpg 10
polyu\p10\s10_p10_c3.jpg 10
polyu\p10\s10_p10_c4.jpg 10
polyu\p11\s10_p11_c1.jpg 11
polyu\p11\s10_p11_c2.jpg 11
polyu\p11\s10_p11_c3.jpg 11
polyu\p11\s10_p11_c4.jpg 11
polyu\p12\s10_p12_c1.jpg 12
polyu\p12\s10_p12_c2.jpg 12
polyu\p12\s10_p12_c3.jpg 12
polyu\p12\s10_p12_c4.jpg 12
polyu\p13\s10_p13_c1.jpg 13
polyu\p13\s10_p13_c2.jpg 13
polyu\p13\s10_p13_c3.jpg 13
polyu\p13\s10_p13_c4.jpg 13
polyu\p14\s10_p14_c1.jpg 14
polyu\p14\s10_p14_c2.jpg 14
polyu\p14\s10_p14_c3.jpg 14
polyu\p14\s10_p14_c4.jpg 14
polyu\p15\s10_p15_c1.jpg 15