본문 바로가기
Error Correction

cv2.error: OpenCV(4.7.0) D:\a\opencv-python\opencv-python\opencv\modules\imgproc\src\colormap.cpp:736: error: (-5:Bad argument) cv::ColorMap only supports source images of type CV_8UC1 or CV_8UC3 in function 'cv::colormap::ColorMap::operator ()' cv2.app..

by apsdfjoi 2023. 6. 15.
728x90
반응형

Grad Cam 구현 중에 원본 이미지에 히트맵을 그리는 코드는 matplotlib을 사용하여 작성했다.

 

[Grad Cam] gradcam pytorch 구현 with chatGPT

chatGPT를 사용하여 Grad Cam을 구현해 봤다. 물론 한 번에 구현되진 않는다. Grad Cam은 마지막 컨볼루션 레이어를 사용해야 의미있는 결과를 얻을 수 있다. 코드는 다음과 같다. import torch import torch.nn

yeeca.tistory.com

이를 opencv 코드로 옮기면서 에러가 발생했다.

gc_jet = cv2.applyColorMap(result, cv2.COLORMAP_JET)

여기서 잘못된 것인데

result 변수의 shape는 (224,224)였으며 dtype은 float32였다. 

data 값을 0~255, dtype은 uint8로 바꾸면 해결된다.

result = (result * 255).astype(np.uint8)
gc_jet = cv2.applyColorMap(result, cv2.COLORMAP_JET)

 

728x90
반응형

댓글