Update yolov5.py

yolov5.py

@@ -213,7 +213,7 @@ def dff_nmf(image, target_lyr, n_components):
         boxes = boxes[confidence_mask]  # Filter boxes by confidence mask
         fig, ax = plt.subplots(1, figsize=(8, 8))
         ax.axis("off")
-        ax.imshow(torch.tensor(batch_explanations[0][indx]).cpu().numpy(), cmap="Wistia")  # Display image
+        ax.imshow(torch.tensor(batch_explanations[0][indx]).cpu().numpy(), cmap="plasma")  # 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()