From 70313eb750ee5387803f3245856829d13a7edf11 Mon Sep 17 00:00:00 2001
From: LorenzLamm <34575029+LorenzLamm@users.noreply.github.com>
Date: Thu, 27 Jul 2023 12:41:32 +0200
Subject: [PATCH] Fix voxel size loading (#28)

---
 src/membrain_seg/segmentation/dataloading/data_utils.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/membrain_seg/segmentation/dataloading/data_utils.py b/src/membrain_seg/segmentation/dataloading/data_utils.py
index 1e4affd..454ebef 100644
--- a/src/membrain_seg/segmentation/dataloading/data_utils.py
+++ b/src/membrain_seg/segmentation/dataloading/data_utils.py
@@ -241,6 +241,7 @@ 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)
@@ -248,10 +249,10 @@ def load_tomogram(
 
     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