728x90 반응형 GradCam2 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.. 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였.. 2023. 6. 15. [Grad Cam] gradcam pytorch 구현 with chatGPT chatGPT를 사용하여 Grad Cam을 구현해 봤다. 물론 한 번에 구현되진 않는다. Grad Cam은 마지막 컨볼루션 레이어를 사용해야 의미있는 결과를 얻을 수 있다. 코드는 다음과 같다. import torch import torch.nn as nn import torchvision.transforms as transforms import PIL.Image as Image import cv2 import matplotlib.pyplot as plt import numpy as np import os os.environ['KMP_DUPLICATE_LIB_OK'] = 'True' class GradCam: def __init__(self, model, target_layer): self.model =.. 2023. 6. 14. 이전 1 다음 728x90 반응형
