convert to numpy before showing image

ETSIhackers · Nov 17, 2023 · a6d75f3 · a6d75f3
1 parent c50b3e1
commit a6d75f3
Showing 1 changed file with 1 addition and 1 deletion.
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")