Fix voxel size loading (#28)

teamtomo · Jul 27, 2023 · 70313eb · 70313eb
1 parent a5cece6
commit 70313eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/membrain_seg/segmentation/dataloading/data_utils.py b/src/membrain_seg/segmentation/dataloading/data_utils.py
@@ -241,17 +241,18 @@ def load_tomogram(
         data = tomogram.data.copy()
         data = np.transpose(data, (2, 1, 0))
         header = tomogram.header
+        voxel_size = tomogram.voxel_size
     if normalize_data:
         data = img_as_float32(data)
         data -= np.mean(data)
         data /= np.std(data)
 
     if return_header:
         if return_pixel_size:
-            return data, header, tomogram.voxel_size
+            return data, header, voxel_size
         return data, header
     if return_pixel_size:
-        return data, tomogram.voxel_size
+        return data, voxel_size
     return data