[cv2] image input 3-channel Grayscale

딥러닝 모델 input이 3channel (RGB)기반 모델인데, Grayscale로 inference/validate을 진행하고 싶을 때는 기존 RGB Load방식에서 아래와 같이 1 channel로 로드하고 동일한 layer를 3겹 쌓아서 넘겨주면 됨.

 

맨날 함수를 까먹어서 기록용.

im = cv2.imread(f, cv2.IMREAD_GRAYSCALE)
if len(im.shape) <3:# grayscale
im = np.dstack([im, im, im])