[Programming / Python] 이미지를 픽셀 데이터로 가져오기

Python의 이미지 라이브러리인 pillow를 활용하여 이미지 데이터를 불러온 후, 이 이미지의 픽셀 데이터를 numpy 라이브러리의 array로 저장하는 코드는 다음과 같습니다:

import PIL.Image as pilimg
import numpy as np
 
# Read image
im = pilimg.open( {YOUR_IMAGE_PATH} )
 
# Display image
im.show()
 
# Fetch image pixel data to numpy array
pix = np.array(im)

만약 matplotlib 라이브러리를 활용하여 이미지를 디스플레이 하고자 한다면 다음 코드를 이용하시면 됩니다:

import PIL.Image as pilimg
import numpy as np
import matplotlib.pyplot as plt
 
# Read image
im = pilimg.open( {YOUR_IMAGE_PATH} )
 
# Fetch image pixel data to numpy array
pix = np.array(im)
 
plt.imshow(pix)

단, 불러온 이미지가 Greyscale로 표시되어야 할 경우, 다음 코드를 활용하면 됩니다:

plt.imshow(pix, cmap='gray')