From 8d4e234dd904a9fa00595faa93dfe8f7bb793e60 Mon Sep 17 00:00:00 2001 From: Sam Freesun Friedman Date: Fri, 24 Jan 2025 11:05:01 -0500 Subject: [PATCH] compute ms ssim in test steps --- ml4h/tensormap/ukb/mri_brain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml4h/tensormap/ukb/mri_brain.py b/ml4h/tensormap/ukb/mri_brain.py index 1ecf75576..b0f2fa398 100755 --- a/ml4h/tensormap/ukb/mri_brain.py +++ b/ml4h/tensormap/ukb/mri_brain.py @@ -451,7 +451,7 @@ def random_mni_slice_tensor(tm, hd5, dependents={}): tm.dependent_map.shape, dtype=np.float32, ) - dependents[tm.dependent_map][0] = float(slice_index) / 182.0 + dependents[tm.dependent_map][0] = (float(slice_index) - 96) / 8.0 return tensor return random_mni_slice_tensor