Python:读取文件夹下的图像或其他文件后，保存到列表并输出到txt中。

这个程序是在用faster进行批量测试的时候写的，首先把要测试的图像放到一个文件夹下面，然后用程序读取名字保存到test.txt中，在运行faster检测程序的时候，读取这个列表，修改检测程序，按列表里的名字对图像一一进行检测就行了。

"""
https://blog.csdn.net/gusui7202/article/details/83239142
qhy。
"""
import sys
import os 
data_dir = "/home/q/tf-faster-rcnn-master/data/demo/"
file_list = [] 
write_file_name = '/test.txt'
write_file = open(write_file_name, "w")
tt=os.listdir(data_dir)
for file in tt:
    if file.endswith(".jpg"):
        write_name = file
        file_list.append(write_name)
print(file_list)
for current_line in range(len(file_list)): 
     write_file.write(file_list[current_line] + '\n')
write_file.close()