흑백 이미지 데이터 셋에 관련된 질문입니다.

안녕하세요. 선생님

정말 질 좋은 강의를 들을 수 있어서 행운이라고 생각합니다.

CatnDog_Sequence기반으로 개인적으로 Modify하여 공부를 해보고 있습니다.

제가 갖고 있는 이미지는 흑백 이미지 인데요.

image = cv2.imread("D:\\Lens_image_data\\test\\test_set\\sort1\\1.BMP", cv2.IMREAD_ANYCOLOR)
print(image.shape)

shape을 print해보았을 때, (1548, 1544)으로 RGB 3채널이 아닌 것도 확인하였습니다.

1) 따라서 Image_Batch 배열 진행 시에도 뒤에 3채널은 빼고 넣었으며,

image_batch = np.zeros((image_name_batch.shape[0], IMAGE_SIZE, IMAGE_SIZE), dtype='float32')

2) 이미지를 cv2로 읽어올 때도 아래와 같이 변경해보았습니다.

image = cv2.imread(image_name_batch[image_index], cv2.IMREAD_UNCHANGED)

이렇게 되면 shape이 (64, 224, 224) (64,) 이렇게 나오는데, 흑백 이미지로 할 때는 상기와 같이 바꾸는게 맞는 것인지 궁금합니다.

            if self.pre_func is not None:
                image = self.pre_func(image)

            image_batch[image_index] = image
            
        image_batch = np.repeat(image_batch[..., np.newaxis], 3, -1)
        return image_batch, label_batch