Debug: yolov8 target lyr

yolov8.py

@@ -47,8 +47,15 @@ def xai_yolov8(image):
     # Convert image to PyTorch tensor for Grad-CAM
     image_tensor = torch.tensor(np.array(image)).permute(2, 0, 1).unsqueeze(0).float() / 255.0
     image_tensor = image_tensor.to('cpu')
+    
+    # Inspect the model and print layer structure
     print(model.model)  # Output model layers to find the target layer
-    grad_cam = GradCAM(model.model, target_layer='model.model[-1]')  
+    
+    # Find the correct target layer
+    target_layer = model.model.model[-1]  # Target the last detection layer
+    
+    # Initialize Grad-CAM with the target layer
+    grad_cam = GradCAM(model.model, target_layer=target_layer)
 
     # Perform Grad-CAM
     cam_map = grad_cam(image_tensor)
@@ -59,4 +66,5 @@ def xai_yolov8(image):
     # Combine original image and Grad-CAM image
     final_image = np.hstack((np.array(image), cam_image))
     caption = "Results using YOLOv8 and Grad-CAM via torchcam"
+    
     return Image.fromarray(final_image), caption