Grayscale images to 3 channels for CNN

ResNet이나 ViT와 같은 대용량의 데이터로 pre-train된 모델을 적용하는 경우가 있다.

CNN 기반의 이 모델들은 칼라 이미지를 사용했기 때문에 data shape은 (m, n, 3)이 된다.

그런데 내가 가진 데이터가 흑백이라면 이 모델에 어떻게 넣어야 할까?

 

skimage에 있는 gray2rgb를 이용하면 3채널이 된다.

pip install scikit-image
from skimage.color import gray2rgb