Implement Visual Comparison Utility

Author: AAdewunmi

notebooks/04_gradcam_refinement.ipynb

@@ -134,6 +134,34 @@
     "    heatmap = heatmap.astype(np.float32) / 255.0\n",
     "    return heatmap"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "6bda37ff",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# ------------------------------------------------------------\n",
+    "# 3. Visual Comparison Utility\n",
+    "# ------------------------------------------------------------\n",
+    "def compare_models(image_path: Path, model_paths: list):\n",
+    "    \"\"\"Compare Grad-CAM results from multiple model checkpoints.\"\"\"\n",
+    "    results = []\n",
+    "    for path in model_paths:\n",
+    "        heatmap = generate_cam(image_path, path)\n",
+    "        img = cv2.imread(str(image_path))\n",
+    "        overlay = GradCAM.overlay_heatmap(heatmap, img)\n",
+    "\n",
+    "        out_name = f\"{image_path.stem}_{path.stem}_overlay.png\"\n",
+    "        out_path = output_dir / out_name\n",
+    "        cv2.imwrite(str(out_path), overlay)\n",
+    "\n",
+    "        results.append((path.stem, heatmap, overlay))\n",
+    "        print(f\"Saved overlay: {out_path.name}\")\n",
+    "\n",
+    "    return results"
+   ]
   }
  ],
  "metadata": {