python 提取jpg文件名字到txt

python 提取jpg文件名字到txt

# -*- coding:utf-8 -*-
import os
# 图片地址
data_base_dir = r"C:\Users\HP\Desktop\SRNet-eng\SRNet-Datagen\dataset/bg"
file_list = [] #建立列表，用于保存图片信息
# txt文件地址
write_file_name = r'C:\Users\HP\Desktop\SRNet-eng\SRNet-Datagen\dataset/bg/labels.txt'
write_file = open(write_file_name, "w") #以只写方式打开write_file_name文件
for file in os.listdir(data_base_dir): #file为current_dir当前目录下图片名
    if file.endswith(".jpg"): #如果file以jpg结尾
        write_name = file #图片路径 + 图片名 + 标签
        file_list.append(write_name) #将write_name添加到file_list列表最后
        sorted(file_list) #将列表中所有元素随机排列
        number_of_lines = len(file_list) #列表中元素个数
#将图片信息写入txt文件中
for current_line in range(number_of_lines):
    write_file.write(file_list[current_line] + '\n')
#关闭文件
write_file.close()