将图片路径与标签路径拼接写入到txt文档

from pathlib import Path
if __name__ == '__main__':

    img_files = [
        Path("C:/Users/ytc79/Desktop/CrowdCounting-P2PNet-main/vis/part_B_final/train_data/images/")]#图片路径

    txt_files = [
        Path("C:/Users/ytc79/Desktop/CrowdCounting-P2PNet-main/vis/part_B_final/train_data/txt/")]#txt路径
    result_png_list = r"C:/Users/ytc79/Desktop/CrowdCounting-P2PNet-main/vis/part_B_final/train_data/train.txt"#要写到哪个文件中

    write_png = open(result_png_list, "w", encoding="utf-8")#打开要写入的txt文件
    for img_file, txt_file in zip(img_files, txt_files):
        for img_pth, txt_pth in list(zip(img_file.glob("IMG_*.jpg"), txt_file.glob("GT_IMG_*.txt"))):
        #我这里是IMG_开头的jpg和GT_IMG_开头的txt对应,如果只有后缀名不同,直接img_file.glob("*.jpg"), txt_file.glob("*.txt")即可
            write_png.write(str(img_pth) + "\x20" + str(txt_pth)+'\n')#路径拼接,加一个空格\x20,,再换行

    write_png.close()#关闭文件


结果如图所示:
在这里插入图片描述

参考了这位老师的代码:https://blog.csdn.net/xiangzun0155/article/details/128470042

猜你喜欢

转载自blog.csdn.net/weixin_46504385/article/details/130455040