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
반응형
댓글