파이썬 rgb 색상별 추출 pillow image.split

파이썬 rgb 색상별 추출¶

• pillow를 활용하여 r,g,b 값을 간단히 분류 하는 방법을 소개합니다.

In [1]:

    import numpy as np
    from PIL import Image
    import matplotlib.pyplot as plt
    img = Image.open("test_image.png")
    # 한눈에 들어오는 이미지를 테스트 대상으로 사용하기 위해 이미지 resize를 합니다.
    img = img.resize((256, 248))

In [2]:

    # split 함수를 통해 rgb 색상을 간단히 추출할 수 있습니다.
    r, g, b = img.split()

In [3]:

    # 넘파이 배열로 r 값을 변환하고 g, b, 차원의 값을 0으로 채우면 r 값만 반영된 색상을 확인 할 수 있습니다.
    r_array = np.asarray(r)
    empty_array = np.zeros((248, 256, 3))
    empty_array[:,:,0] = r_array

In [4]:

    # r 값이 잘 추출 되었음을 알 수 있습니다.
    Image.fromarray(np.uint8(empty_array))

Out[4]:

In [5]:

    # 분리된 r, g, b 변수를 merge 함수를 이용하여 원래 상태로 돌릴 수 있습니다.
    img = Image.merge("RGB", (r, g, b))
    img

Out[5]: