compute ms ssim in test steps

broadinstitute · Jan 24, 2025 · 8d4e234 · 8d4e234
1 parent 7e58456
commit 8d4e234
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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