pgm revert转换 成jpg 人脸识别图片

最近在搞人脸识别，下载数据集走得比较心累。很多数据集太大了。没有啥标签。先搞一个小的玩玩。还找到的是pgm灰度图。索性写了个小脚本，用来转换。同时写脚本打标签。

数据集地址：http://download.csdn.net/detail/u014609362/7804183#comment

代码附上。

from PIL import Image
import os
import shutil
import re
textfile=''
filex=open(textfile,'w+')
def main():
dir='/'
output=''
for filepath in os.listdir(dir):
filename=os.path.join(dir+filepath)
name = os.path.basename(filename)
tmp=name.split('-')
if tmp[0]=='w':
label=str(1)+tmp[1]
else:
print "tmp[0] is" +tmp[0]
print "tmp[1] is" +tmp[1]

label=str(2)+tmp[1]
tname=name.split('.')
newname=tname[0]+".JPG"
filex.write(newname+" "+label)
filex.write('\n')
img = Image.open(filename)
# output=os.mkdir(os.path.join(dir+"newdir"))
# os.path.join(dir+newname)
img.save(os.path.join(output+newname))

# shutil.
if __name__ == '__main__':
main()
filex.close()