生成txt

import os
'''
全部生成txt文档并不区分测试集和训练集
'''
path = "C:\\Users\Administrator\Desktop\SAR_specific_models-master\data\\spe_data"  # 图片集路径

classes = [i for i in os.listdir(path)]

cls = open("clst.txt", 'w')
for cla in classes:
    s = 0
    for img in os.listdir(os.path.join(path, cla)):
        # 此处os.listdir(os.path.join(path, i))读取到了i类别里面的image
        name_cla = os.path.join(path, cla) + '\\' + img + ',' + cla + '\n'
        cls.write(name_cla)
        s += 1
cls.close()