os.walk

import os

# 定义文件夹名称和标签的对应关系
label_map = {
    'cat': 0,
    'dog': 1,
    'bat': 2
}

with open('data.txt', 'w') as f:

    # 遍历所有文件，root为当前文件夹，dirs是所有子文件夹名，files是所有文件名
    for root, dirs, files in os.walk('data'):
        for filename in files:
            # 获得文件完整路径
            filepath = os.path.join(root, filename)
            # 获取当前文件夹名称,即类别名
            dirname = os.path.split(root)[-1]
            # 得到类别名对应的标签
            label = label_map[dirname]
            # 将 图片路径+标签 存入data.txt文件中
            f.write('{},{}\n'.format(filepath, label))

图片路径，标签