diff --git a/direct/data/datasets.py b/direct/data/datasets.py index 12c0f1d0..728c1e1d 100644 --- a/direct/data/datasets.py +++ b/direct/data/datasets.py @@ -180,7 +180,6 @@ def __getitem__(self, idx: int) -> dict[str, Any]: # Get the k-space absolute value and create the mask. kspace_abs = sample["kspace"].sum(0) sample["sampling_mask"] = np.mean(np.abs(kspace_abs).sum(0)) < np.abs(kspace_abs).sum(0) - sample["sampling_mask"] = sample["sampling_mask"] sample["kspace"] = sample["kspace"] * sample["sampling_mask"][None]