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

原创于 2019-01-10 17:14:42 发布 · 1w 阅读

31 ·

CC 4.0 BY-SA版权

文章标签：

#python #人工智能 #文件处理

python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何使用Python生成YOLOv3所需的所有图片路径的txt文件，通过遍历指定文件夹，获取JPEGImages下所有图片的绝对路径并写入train_all.txt，适用于自定义数据集的训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、介绍

在做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文件。