diff --git a/direct/common/subsample.py b/direct/common/subsample.py index d3ea52a7..92c3b3c6 100644 --- a/direct/common/subsample.py +++ b/direct/common/subsample.py @@ -310,7 +310,6 @@ def mask_func( num_cols = shape[-2] num_center_lines, acceleration = self.choose_acceleration() - print(num_center_lines, acceleration) mask = self.center_mask_func(num_cols, num_center_lines) diff --git a/direct/data/mri_transforms.py b/direct/data/mri_transforms.py index 0208ad22..6a6a536d 100644 --- a/direct/data/mri_transforms.py +++ b/direct/data/mri_transforms.py @@ -764,7 +764,7 @@ def __call__(self, sample: Dict[str, Any], coil_dim: int = 0) -> Dict[str, Any]: seed = None if not self.use_seed else tuple(map(ord, str(sample["filename"]))) kspace_shape = tuple(sample["kspace"].shape[-3:]) acs_mask = self.mask_func(shape=kspace_shape, seed=seed, return_acs=True) - print(acs_mask.shape) + kspace = acs_mask * kspace + 0.0 dim = self.spatial_dims["2D"] if kspace.ndim == 4 else self.spatial_dims["3D"] acs_image = self.backward_operator(kspace, dim=dim)