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()
生成txt
最新推荐文章于 2024-10-02 10:53:34 发布