处理数据集——将label及路径保存在txt文件中

import os

def traverse_dir(path):
for root, dirs, files in os.walk(path):
# print(“当前目录：”, root)
# print(“子目录列表：”, dirs)
# print(“文件列表：”, files)
for file in files:
element = os.path.join(root, file)
print(element)
with open(‘D:\codeagain\datasets\AF7\output.txt’, ‘a’,newline=‘’) as file:
file.write(element + ‘\n’)

dir_path = “D:\codeagain\datasets\AF7”
print(‘待遍历的目录为：’, dir_path)
print(‘遍历结果为：’)
traverse_dir(dir_path)