Fix support for vmin/vmax when plotting

fepegar · fepegar · commit 67d13defb9bd · 2025-09-17T22:28:54.000+01:00
diff --git a/src/torchio/visualization.py b/src/torchio/visualization.py
@@ -146,8 +146,10 @@ def plot_volume(
             ]
         )
         p1, p2 = np.percentile(displayed_data, percentiles)
-        imshow_kwargs['vmin'] = p1
-        imshow_kwargs['vmax'] = p2
+        if 'vmin' not in imshow_kwargs:
+            imshow_kwargs['vmin'] = p1
+        if 'vmax' not in imshow_kwargs:
+            imshow_kwargs['vmax'] = p2
 
     spacing_r, spacing_a, spacing_s = image.spacing
     sag_axis, cor_axis, axi_axis = axes

Original file line number	Diff line number	Diff line change
`@@ -146,8 +146,10 @@ def plot_volume(`
`146`	`146`	`]`
`147`	`147`	`)`
`148`	`148`	`p1, p2 = np.percentile(displayed_data, percentiles)`
`149`		`- imshow_kwargs['vmin'] = p1`
`150`		`- imshow_kwargs['vmax'] = p2`
	`149`	`+ if 'vmin' not in imshow_kwargs:`
	`150`	`+ imshow_kwargs['vmin'] = p1`
	`151`	`+ if 'vmax' not in imshow_kwargs:`
	`152`	`+ imshow_kwargs['vmax'] = p2`
`151`	`153`
`152`	`154`	`spacing_r, spacing_a, spacing_s = image.spacing`
`153`	`155`	`sag_axis, cor_axis, axi_axis = axes`