Fix Density comp

CEA-COSMIC · Jan 10, 2024 · 605eab2 · 605eab2
1 parent ad4ecff
commit 605eab2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 79 deletions.
diff --git a/examples/GPU_Examples/NonCartesian_gpuNUFFT_DensityCompensation.py b/examples/GPU_Examples/NonCartesian_gpuNUFFT_DensityCompensation.py
diff --git a/mri/operators/fourier/non_cartesian.py b/mri/operators/fourier/non_cartesian.py
@@ -56,6 +56,7 @@ def __init__(self, samples, shape, implementation='finufft', n_coils=1,
         self.impl = get_operator(self.implementation)(
             self.samples,
             self.shape,
+            density=self.density_comp,
             **self.kwargs
         )
 

diff --git a/mri/operators/fourier/utils/processing.py b/mri/operators/fourier/utils/processing.py
@@ -285,4 +285,4 @@ def estimate_density_compensation(kspace_loc, volume_shape, implementation='pipe
         volume_shape,
         **kwargs
     )
-    return density_comp
+    return np.squeeze(density_comp)