Change name in visualization of heatmap

scumatteo · Jan 18, 2021 · 2d8c952 · 2d8c952
1 parent d2dac13
commit 2d8c952
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/heatmap_visualization.py b/src/heatmap_visualization.py
@@ -80,9 +80,9 @@ def __overlay_gradCAM(img, cam3):
 def __vis_heatmap(cam, cam3, heatmap):
     fig, ax = plt.subplots(1, 3, figsize=(15,15))
     ax[0].imshow(cam)
-    ax[0].set_title("HeatMap")
+    ax[0].set_title("Cam")
     ax[1].imshow(cam3)
-    ax[1].set_title("HeatMap Relu")
+    ax[1].set_title("Cam Relu")
     ax[2].imshow(heatmap[0])
     ax[2].set_title("Image with HeatMap")
     plt.tight_layout()