update: heatmap change

yolov5.py

@@ -212,7 +212,7 @@ def dff_nmf(image, target_lyr, n_components):
         boxes = output1[..., :4]  # First 4 values are x1, y1, x2, y2
         boxes = boxes[confidence_mask]  # Filter boxes by confidence mask
         fig, ax = plt.subplots(1, figsize=(10, 10))
-        ax.imshow(torch.tensor(batch_explanations[0][indx]).cpu().numpy(), cmap="RdYIGn")  # Display image
+        ax.imshow(torch.tensor(batch_explanations[0][indx]).cpu().numpy(), cmap="RdYlGn")  # Display image
         top_score_idx = scores.argmax(dim=0)  # Get the index of the max score
         top_score = scores[top_score_idx].item()
         top_class_id = class_ids[top_score_idx].item()