- <span style="font-size:14px;">
- #!/usr/bin/python3 ''' Created on Jul 16, 2016 @author: pang ''' import
- os #input files' path ROOT_DIR='/home/knaffe/workspace/caffe-master/data/deepiddata/train'##train文件夹的路径,如果要修改成其他,请注意这个路径的格式,caffe-master/data/deepiddata/train
- #output files' path OUTPUT_DIR='./pair.txt'###生成的文件名。存放路径为/home/knaffe/workspace/caffe-master/data/deepiddata/
- FORMAT='jpg'###用来排除其他文件干扰,若读取到其他文件, caffe转化格式脚本create_imagenet.sh会报错。 if
- __name__=='__main__': if(ROOT_DIR.endswith('/') is not True): ROOT_DIR+='/'
- file=os.listdir(path=ROOT_DIR) label=0 with open(OUTPUT_DIR,'w') as f:
- for sub_file in file: imglist=os.listdir(ROOT_DIR+sub_file) imglist for
- img in imglist: if(img.endswith(FORMAT) is not True): continue str_tmp=str(sub_file)+'/'+img+'
- '+str(label) f.write(str_tmp+'\n') label+=1 print('Finish Processing')
- </span>
来源: http://lib.csdn.net/snippet/deeplearning/45575