폴 더 의 그림 을 파일 이름 에 따라 태그 로 분류 합 니 다.

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)+'
'
) i=i+1

좋은 웹페이지 즐겨찾기