Plot Visual Comparison

Author: AAdewunmi

notebooks/04_gradcam_refinement.ipynb

@@ -162,6 +162,34 @@
     "\n",
     "    return results"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "01752cc6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# ------------------------------------------------------------\n",
+    "# 4. Plot Visual Comparison\n",
+    "# ------------------------------------------------------------\n",
+    "def plot_gradcam_comparison(image_path: Path, results):\n",
+    "    \"\"\"Plot baseline vs fine-tuned Grad-CAM overlays side by side.\"\"\"\n",
+    "    fig, axes = plt.subplots(1, len(results) + 1, figsize=(12, 4))\n",
+    "\n",
+    "    original = np.array(Image.open(image_path))\n",
+    "    axes[0].imshow(original)\n",
+    "    axes[0].set_title('Original Image')\n",
+    "    axes[0].axis('off')\n",
+    "\n",
+    "    for i, (label, heatmap, overlay) in enumerate(results, start=1):\n",
+    "        axes[i].imshow(cv2.cvtColor(overlay, cv2.COLOR_BGR2RGB))\n",
+    "        axes[i].set_title(label.replace('_', ' ').title())\n",
+    "        axes[i].axis('off')\n",
+    "\n",
+    "    plt.tight_layout()\n",
+    "    plt.show()"
+   ]
   }
  ],
  "metadata": {