파이썬 이미지 불러오기, 저장하기(pillow, pil)

파이썬 이미지 불러오기, 저장하기(pillow, pil)¶

• PIL패키지 > Image객체 > open 함수를 사용합니다.

In [1]:

# 패키지 불러오기
from PIL import Image

# 파일이름 변수화
file_name = "test_image.jpeg"

# 이미지 불러오기
img = Image.open(file_name)
img

Out[1]:

• PIL를 이용해 불러온 img 객체에서 save 함수를 이용하면 원하는 이름으로 이미지 파일을 저장 할 수 있습니다.

In [2]:

# 이미지 저장하기(jpeg)
img.save("test_image.jpeg")

[참고 1] 머신러닝, 딥러닝 학습을 위해 pil로 불러온 이미지를 array로 전환해야 할 수도 있습니다. 아래는 넘파이를 이용하여 array로 전환하는 코드 입니다.

In [3]:

import numpy as np
# 어레이화
a = np.array(img)

# 어레이의 크기
np.shape(a)

Out[3]:

(1080, 1440, 3)

[참고 2] 딥러닝, 머신러닝 학습을 위해 이미지 데이터를 불러올 때, opencv 패키지를 활용하기도 합니다.

In [5]:

import cv2

# rgb 이미지 불러오기
rgb_image = cv2.imread(file_name)

In [6]:

rgb_image

Out[6]:

array([[[187, 198, 206],
        [186, 197, 205],
        [186, 197, 205],
        ...,
        [154, 163, 173],
        [150, 159, 169],
        [146, 155, 165]],

       [[185, 196, 204],
        [185, 196, 204],
        [185, 196, 204],
        ...,
        [154, 163, 172],
        [154, 160, 171],
        [149, 158, 167]],

       [[183, 194, 202],
        [183, 194, 202],
        [183, 194, 202],
        ...,
        [155, 163, 170],
        [158, 163, 172],
        [156, 164, 171]],

       ...,

       [[131, 119, 109],
        [128, 116, 106],
        [125, 113, 103],
        ...,
        [ 88,  75,  73],
        [ 86,  73,  71],
        [ 82,  69,  67]],

       [[129, 117, 107],
        [129, 117, 107],
        [127, 115, 105],
        ...,
        [ 86,  73,  71],
        [ 84,  71,  69],
        [ 82,  69,  67]],

       [[123, 111, 101],
        [125, 113, 103],
        [127, 115, 105],
        ...,
        [ 85,  72,  70],
        [ 85,  72,  70],
        [ 84,  71,  69]]], dtype=uint8)