fix grid slice

ANTsX · May 15, 2024 · b6dae36 · b6dae36
1 parent 631974b
commit b6dae36
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/ants/viz/plot_grid.py b/ants/viz/plot_grid.py
@@ -158,17 +158,17 @@ def reorient_slice(x, axis):
 
     def slice_image(img, axis, idx):
         if axis == 0:
-            return img[idx, :, :]
+            return img[idx, :, :].numpy()
         elif axis == 1:
-            return img[:, idx, :]
+            return img[:, idx, :].numpy()
         elif axis == 2:
-            return img[:, :, idx]
+            return img[:, :, idx].numpy()
         elif axis == -1:
-            return img[:, :, idx]
+            return img[:, :, idx].numpy()
         elif axis == -2:
-            return img[:, idx, :]
+            return img[:, idx, :].numpy()
         elif axis == -3:
-            return img[idx, :, :]
+            return img[idx, :, :].numpy()
         else:
             raise ValueError("axis %i not valid" % axis)