728x90 반응형 ResNet501 [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 반응형
