code: def __init__(
      self, model, target_layers, reshape_transform=None, use_cuda=False
  ):
      self.model = model.eval()
      self.target_layers = target_layers
      self.reshape_transform = reshape_transform
      self.cuda = use_cuda
      if self.cuda:
          self.model = model.cuda()
      self.activations_and_grads = ActivationsAndGradients(
          self.model, target_layers, reshape_transform
      )
  
imports: import cv2
  import numpy
label_desc: Use image node_id 1 to create a new GradCAM object  with example: obj = GradCAM(model, target_layers, reshape_transform, use_cuda)