Merge pull request #14 from gschramm/main

Bugfix: convert Arrays to numpy before passing them to matplitlib
ETSIhackers · Nov 17, 2023 · e9a1cd6 · e9a1cd6
2 parents dca594d + a6d75f3
commit e9a1cd6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/00_simulate_lm_data.py b/python/00_simulate_lm_data.py
@@ -200,12 +200,12 @@
 )
 for i in range(img_shape[2]):
     ax[0, i].imshow(
-        xp.asarray(to_device(img_f1[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
+        np.asarray(to_device(img_f1[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
     )
     ax[0, i].set_title(f"sl {i+1}", fontsize="small")
 
     ax[1, i].imshow(
-        xp.asarray(to_device(img_f2[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
+        np.asarray(to_device(img_f2[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
     )
 
 ax[0, 0].set_ylabel("ground truth frame 1", fontsize="small")

diff --git a/python/01_reconstruct_lm_data.py b/python/01_reconstruct_lm_data.py
@@ -144,7 +144,7 @@
 fig, ax = plt.subplots(1, recon.shape[2], figsize=(recon.shape[2] * 2, 2))
 for i in range(recon.shape[2]):
     ax[i].imshow(
-        xp.asarray(to_device(recon[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
+        np.asarray(to_device(recon[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
     )
     ax[i].set_title(f"LM recon sl {i+1}", fontsize="small")