Python批量读取人脸图片与数据互相转换

# -*- coding: utf-8 -*-
import numpy as np 
import matplotlib.pyplot as plt
from PIL import Image
def ImagetoData(num=100):
    filename = ‘face/face‘
    data = []
    for i in range(num):
        filename = ‘face/face‘+ str(i) + ‘.jpg‘
        # 读取图片
        im = Image.open(filename)
    # 显示图片
#     im.show()  
        width,height = im.size
        im = im.convert("L") 
        picdata = im.getdata()
        picdata = list(picdata)
        data.append(picdata)
    data = np.matrix(data,‘float‘)/255.0
    return data
     
def DatatoImage(title,Data,savename):
    w,h = np.shape(Data)
    n_col = int(w**0.5)
    n_row = n_col 
    p_size = int(h**0.5)
    plt.figure(figsize=(2*n_col,2.26*n_row))
    plt.suptitle(title,size=16)
    for i in range(w):
        plt.subplot(n_row,n_col,i+1)
        plt.imshow(Data[i,:].reshape((p_size,p_size)),
                   cmap=plt.cm.gray,
                   interpolation=‘nearest‘)
        plt.xticks(())
        plt.yticks(())
    plt.subplots_adjust(0.01, 0.05, 0.99, 0.93, 0.04, 0.)
    plt.savefig(savename)
                  

data = ImagetoData()
title = ‘face‘
savename=‘rawface.jpg‘
DatatoImage(title,data,savename)