python 批量将图片名称保存在txt文件中

python 批量将图片名称保存在txt文件中

在处理图片时，需要将图片的名称全部读取并保存下来，因此记录这个小程序，以备后续使用

---

---

前言

处理数据集时，有时候需要将数据集中图片名称批量读取，因此记录这个小程序

---

以下是本篇文章正文内容，下面案例可供参考

一、批量读取图片名称？

使用的是os库

二、使用步骤

1.代码

代码如下（示例）：

import os

def createFileList(images_path, txt_save_path):
    # 打开图片列表清单txt文件
    fw = open(txt_save_path, "w")
    # 查看文件夹下的图片名称
    images_name = os.listdir(images_path)
    count = 0
    # 遍历所有文件名
    for eachname in images_name:
        # 按照规则将内容写入txt文件中
        #name = os.path.splitext(eachname)
        # print(name)
        #filename = name[0]  
        Newdir = os.path.join( '05' + "_" + str(count).zfill(3) + ".jpg")
        ##这里更改的图片名称Newdir：05_001.jpg,可更改后面的“.jpg”来改变读取的图片格式，如“png”
        fw.write(eachname + '   ' + Newdir +'\n')
        #将读取的图片名称与更改后的图片名称，同时保存在txt文件中
        count += 1
    # 打印成功信息
    print("生成txt文件成功")
    # 关闭fw
    fw.close()


# 下面是相关变量定义的路径
if __name__ == '__main__':
    txt_path = "E:/Huca/LowResolution512/"
    # 图片存放目录
    images_path = 'E:/Huca/风格调色图片/中式-红/转档'
    # 生成的图片列表清单txt文件名
    txt_name = '05.txt'
    # 生成的图片列表清单txt文件的保存目录
    txt_save_path = txt_path + txt_name
    # 生成txt文件
    createFileList(images_path, txt_save_path)

2.图片

结果如下（示例）：
要读取的数据集名称

保存中txt文件中的图片名称，可看到读取的图片名称和更改的图片名称一一对应。

---

总结

学习是一个不断积累的过程