[torchvision] RGB to Grayscale Image

 

- 목차

 

Grayscale .

이번 글에서는 torchvision 모듈을 사용하여 RGB Channel 의 이미지를 Grayscale 이미지로 변형하는 방법에 대해 알아보겠습니다.

Grayscale 이란 RGB 채널을 가지는 이미지를 하나의 Channel 을 가지도록 변형하는 스케일링 방식입니다.

일반적으로 하나의 이미지는 아래와 같이 3개의 채널로 구성됩니다.

각 채널은 Red, Blue, Green 으로 표현되는데요. 이러한 색상의 요소들이 모여서 하나의 이미지의 색상을 결정합니다.

출처 : https://www.pexels.com/photo/close-up-photo-of-sitting-puppy-1458926/

 

Grayscale 은 RGB 채널의 컬러를 Black - White 범위를 가지는 Grayscale 로 컬러를 변형합니다.

그 결과로써 아래의 강아지 이미지는 RGB 컬러에서 흑백 색상으로 변경됩니다.

출처 : https://www.pexels.com/photo/close-up-photo-of-sitting-puppy-1458926/

 

아래의 코드는 위의 사진의 결과를 생성하는 Python 코드입니다.

import torchvision
from PIL import Image
import matplotlib.pyplot as plt

gray_scaler = torchvision.transforms.Grayscale()

origin_image = Image.open("../images/pexels-poodles2doodles-1458926.jpg")
plt.imshow(origin_image)
plt.show()

gray_image = gray_scaler(origin_image)
plt.axis("off")
plt.imshow(gray_image, cmap='gray')
plt.show()