将文件夹中的图片按照文件名进行标签分类

import os
import sys
path="C:/flower_classification/train"
filenames=os.listdir(path) #读取path内所有文件名返回列表
i=0
with open("C:/flower_classification/train.txt","w") as f:
        for filename in filenames:
                sub_path=path+"/"+filename+"/"
                pic=os.listdir(sub_path)
                #print(filename+','+str(i))
                for p in pic:
                        f.write(p+','+'c'+str(i)+'\n')
                i=i+1