[Python] image 파일 numpy.ndarray로 불러오기

image 파일을 numpy.ndarray로 불러오는 방법은 3가지가 있다.

OpenCV

$ pip install opencv-python-headless

import cv2

img_cv = cv2.imread('./testimg.jpg')
print(img_cv)

[[[ 21  48  92]
  [ 19  46  90]
  [ 16  42  88]
  ...
  [  0  14  13]
  [  0  11  10]
  [  0  15  14]]

 [[ 17  44  88]
...

OpenCV의 imread는 BGR (Blue, Green, Red) 순서의 numpy.ndarray를 리턴한다.

matplotlib.image

$ pip install matplotlib

import matplotlib.image as matimg

img_mat = matimg.imread('./testimg.jpg')
print(img_mat)

[[[ 92  48  21]
  [ 90  46  19]
  [ 88  42  16]
  ...
  [ 13  14   0]
  [ 10  11   0]
  [ 14  15   0]]

 [[ 88  44  17]
...

matplotlib.image.imread는 RGB 순서의 numpy.ndarray를 리턴한다.

Pillow

$ pip install pillow

import numpy as np
from PIL import Image

img_pil = Image.open('./testimg.jpg')
print(type(img_pil)

img_pil_arr = np.array(img_pil)
print(img_pil_arr)

<class 'PIL.JpegImagePlugin.JpegImageFile'>
[[[ 92  48  21]
  [ 90  46  19]
  [ 88  42  16]
  ...
  [ 13  14   0]
  [ 10  11   0]
  [ 14  15   0]]

 [[ 88  44  17]

Pillow는 Image.open을 이용해 image 파일을 읽어들이는데 리턴 값이 Image 타입 객체이기 때문에 numpy.array를 이용해 numpy.ndarray로 변환시켜 주어야 한다. (순서는 RGB)