[Python] image to csv

from PIL import Image

import numpy as np 
im = Image.open('img2.png') 
 
pixels = list(im.getdata())
pixel_list =[]

print(len(pixels))


myArray = np.array(pixels)
myArray = myArray.astype(int)

np.savetxt("pixel_data.csv", myArray, delimiter=", ", newline=" ",fmt='%i')

 

PIL을 이용하여 이미지를 pixel으로 변환하는 코드.

 

딥러닝 모델에 직접 제작한 이미지 파일을 사용하고 싶은데

모델에서 인풋을 csv로 변경해야하기 때문에 이런 코드가 필요했다.

 

저장 할 때 integer로 저장되고 1칸에 값들이 1차원 리스트로 저장될 수 있도록 하였다.

 

(fer2013.csv랑 같은 포맷으로 만들기 위해서)