文件夹中图像生成txt文本 img2txt

import os#函数def ListFilesToTxt(dir,file,wildcard,recursion):    exts = wildcard.split(" ")    for root, subdirs, files in os.walk(dir):        for name in files:            txt=os.path.splitext(name)            for ext in exts:                if(name.endswith(ext)):                    file.write("/JPEGImages/"+txt[0]+".jpg"+" "+"/SegmentationClassAug/"+txt[0] +".png"+ "\n")                    break        if(not recursion):            breakdef Test():  dir="/～/deeplab1/exper/datavoc/VOC2012/SegmentationClassAug"  outfile="/～/lly/deeplab1/exper/voc12/list/train_aug.txt"  wildcard = ".png"  file = open(outfile,"w")  if not file:    print ("cannot open the file %s for writing" % outfile)  ListFilesToTxt(dir,file,wildcard, 0)  file.close()Test()