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')
效果如下图: