728x90
반응형
딥러닝 모델 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])728x90
반응형
'머신러닝 > Computer Vision' 카테고리의 다른 글
| [cv2] rotated text(box) image cropping (0) | 2023.09.01 |
|---|---|
| [torch, dataset] Image dataloader with path (0) | 2023.08.31 |
| [Python/PIL] 텍스트에 취소선 넣기 (0) | 2023.05.04 |
| PIL Image Font 속성 (0) | 2023.04.27 |
| RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 16 but got size 15 for tensor number 1 in the list. / CV모델 image input size 조정 (0) | 2023.04.18 |
