python 提取jpg文件名字到txt
import os
data_base_dir = r"C:\Users\HP\Desktop\SRNet-eng\SRNet-Datagen\dataset/bg"
file_list = []
write_file_name = r'C:\Users\HP\Desktop\SRNet-eng\SRNet-Datagen\dataset/bg/labels.txt'
write_file = open(write_file_name, "w")
for file in os.listdir(data_base_dir):
if file.endswith(".jpg"):
write_name = file
file_list.append(write_name)
sorted(file_list)
number_of_lines = len(file_list)
for current_line in range(number_of_lines):
write_file.write(file_list[current_line] + '\n')
write_file.close()