BhumikaMak commited on
Commit
200f76c
·
1 Parent(s): cbb7bf6

Debug: yolov8 target lyr

Browse files
Files changed (1) hide show
  1. yolov8.py +1 -1
yolov8.py CHANGED
@@ -48,7 +48,7 @@ def xai_yolov8(image):
48
  image_tensor = torch.tensor(np.array(image)).permute(2, 0, 1).unsqueeze(0).float() / 255.0
49
  image_tensor = image_tensor.to('cpu')
50
  print(model.model) # Output model layers to find the target layer
51
- grad_cam = GradCAM(model.model, target_layer='[model.model.model[-4]]')
52
 
53
  # Perform Grad-CAM
54
  cam_map = grad_cam(image_tensor)
 
48
  image_tensor = torch.tensor(np.array(image)).permute(2, 0, 1).unsqueeze(0).float() / 255.0
49
  image_tensor = image_tensor.to('cpu')
50
  print(model.model) # Output model layers to find the target layer
51
+ grad_cam = GradCAM(model.model, target_layer='model.model.model[-4]')
52
 
53
  # Perform Grad-CAM
54
  cam_map = grad_cam(image_tensor)