readime.py 458 Bytes
Newer Older
Sugon_ldc's avatar
Sugon_ldc committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import pandas as pd
from PIL import Image
import numpy as np

df=pd.read_csv("data/fer2013/fer2013.csv",chunksize=200)
for d in df:
    pixels=d['pixels'].apply(lambda x:x.split(" ")).tolist()
    # pixels=pixels.split(" ")
    pix=pixels[0]
    pix_2arry=[]
    pix = [int(x) for x in pix]
    for i in range(0,len(pix),64):
        pix_2arry.append(pix[i:i+64])

    img = Image.fromarray(np.array(pix_2arry).astype(np.uint8))
    img.show()