Expose imshow kwargs

Author: eigenvivek

diffdrr/visualization.py

@@ -24,6 +24,7 @@ def plot_drr(
     ticks: bool | None = True,
     axs: matplotlib.axes._axes.Axes | None = None,
     cmap: str = "gray",
+    **imshow_kwargs,
 ):
     """Plot an image generated by a DRR module."""
 
@@ -32,7 +33,7 @@ def plot_drr(
     if len(img) == 1:
         axs = [axs]
     for img, ax in zip(img, axs):
-        ax.imshow(img.squeeze().cpu().detach(), cmap=cmap)
+        ax.imshow(img.squeeze().cpu().detach(), cmap=cmap, **imshow_kwargs)
         _, height, width = img.shape
         ax.xaxis.tick_top()
         ax.set(

notebooks/api/04_visualization.ipynb

@@ -79,6 +79,7 @@
     "    ticks: bool | None = True,\n",
     "    axs: matplotlib.axes._axes.Axes | None = None,\n",
     "    cmap: str = \"gray\",\n",
+    "    **imshow_kwargs,\n",
     "):\n",
     "    \"\"\"Plot an image generated by a DRR module.\"\"\"\n",
     "\n",
@@ -87,7 +88,7 @@
     "    if len(img) == 1:\n",
     "        axs = [axs]\n",
     "    for img, ax in zip(img, axs):\n",
-    "        ax.imshow(img.squeeze().cpu().detach(), cmap=cmap)\n",
+    "        ax.imshow(img.squeeze().cpu().detach(), cmap=cmap, **imshow_kwargs)\n",
     "        _, height, width = img.shape\n",
     "        ax.xaxis.tick_top()\n",
     "        ax.set(\n",