Kaggle-Facial Keypoints Detection:原始数据保存为图片

python version: Python 3.6.4

如题,将原始数据“training.csv”中的图片数据保存成文件,直接上代码:

 

import numpy as np
import pandas as pd
from PIL import Image

training = pd.read_csv("training.csv")
training['Image'] = training['Image'].apply(lambda im: np.fromstring(im, sep=' '))

for index, row in training.iterrows():
    img_data = row['Image'].astype('uint8').reshape(96,96)
    img = Image.fromarray(img_data)
    img.save('./Images/' + str(10000000+index) + ".jpg", 'jpeg')

 

效果如下图:

Kaggle-Facial Keypoints Detection:原始数据保存为图片