python写程序自动生成训练集图片的名字

# -*- coding: UTF-8 -*-
import os

#定义caffe根目录
caffe_root = 'E:/caffe/caffe-master/'

#制作训练标签数据
i = 0   #标签
with open(caffe_root + 'examples/test_C/labels/train.txt','w') as train_txt:
    for root,dirs,files in os.walk(caffe_root + 'examples/test_C/data/train/'):#遍历文件夹
        for dir in dirs:
            for root,dirs,files in os.walk(caffe_root + 'examples/test_C/data/train/' + str(dir)):#遍历每一个文件夹中的文件
                for file in files:
                    image_file = str(dir) + '\\' + str(file)
                    label = image_file + ' ' + str(i) + '\n'                #文件路径+空格+标签编号+换行
                    train_txt.writelines(label)                             #写入标签文件中
                i += 1

i = 0   #标签
with open(caffe_root + 'examples/test_C/labels/test.txt','w') as test_txt:
    for root,dirs,files in os.walk(caffe_root + 'examples/test_C/data/test/'):#遍历文件夹
        for dir in dirs:
            for root,dirs,files in os.walk(caffe_root + 'examples/test_C/data/test/' + str(dir)):#遍历每一个文件夹中的文件
                for file in files:
                    image_file = str(dir) + '\\' + str(file)
                    label = image_file + ' ' + str(i) + '\n'                #文件路径+空格+标签编号+换行
                    test_txt.writelines(label)                             #写入标签文件中
                i += 1

print("文本文件生成成功")