【python】读取指定文件夹下的所有图片路径，并保存到列表再输出到txt文件中

一、介绍

在做YOLOv3模型训练自己的数据集时，需要读取所有图片的路径。在YOLOv3中是以txt格式进行读取，所以利用python来生成所有图片路径的txt文件。

二、代码实现

import os
#paths=['coco2014','Stanford','vehicleplate']
paths=['VOC2019']
f=open('train_all.txt', 'w')
for path in paths:
    p=os.path.abspath(path)+'/JPEGImages'
    filenames=os.listdir(p)
    for filename in filenames:
        im_path=p+'/'+filename
        print(im_path)
        f.write(im_path+'\n')
f.close()

其中，VOC2019为存放JEPGImages和labels文件的文件夹，只需要将这个程序放到和VOC2019文件夹同级目录下，再运行这个程序就会生成包含所有图片绝对路径的train_all.txt文件。