读取文件夹下所有文件

# 文件夹路径
img_path = r'E:/workspace/PyCharmProject/dem_feature/dem\512/label'
# txt 保存路径
save_txt_path = r'./images.txt'
# 读取文件夹中的所有文件
imgs = os.listdir(img_path)

过滤：只保留png结尾的图片

# 图片名列表
names = []
# 过滤：只保留png结尾的图片
for img in imgs:
    if img.endswith(".png"):
        names.append(img)

创建txt文件并逐行写入

txt = open(save_txt_path,'w')
for name in names:
    name = name[:-4]    # 去掉后缀名.png
    txt.write(name + '\n')  # 逐行写入图片名，'\n'表示换行
txt.close()

完整代码

"""
#-*-coding:utf-8-*- 
# @author: wangyu a beginner programmer, striving to be the strongest.
# @date: 2022/7/7 20:25
"""
import os
# 文件夹路径
img_path = r'E:/workspace/PyCharmProject/dem_feature/dem/512/label'
# txt 保存路径
save_txt_path = r'./images.txt'
# 读取文件夹中的所有文件
imgs = os.listdir(img_path)
# 图片名列表
names = []
# 过滤：只保留png结尾的图片
for img in imgs:
    if img.endswith(".png"):
        names.append(img)
txt = open(save_txt_path,'w')
for name in names:
    name = name[:-4]    # 去掉后缀名.png
    txt.write(name + '\n')  # 逐行写入图片名，'\n'表示换行
txt.close()