import os
filePath = '/home/xx/data/ECSSD/image'
list_jpg = os.listdir(filePath)
if __name__ == '__main__':
'''
=> 用于完成test.txt, 要在此文件目录下事先创建test.txt文件
与后面完成train.txt分开单独使用时, 直接注释掉另一个
'''
# /=== ===> test.txt 写入pic_name & label <=== ===\
with open('test_pre.txt', 'w') as f: # train.txt或者test.txt 本文件生成在本代码所在位置
for i in list_jpg:
# 如果去掉图片后缀
# i = i.split(".")[0]
f.writelines(i + '\n')
# \=== ===> test.txt 写入pic_name & label <=== ===/
后缀可去可添加