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()