Merge pull request #7 from thorstenwagner/patch-3

Fix computation for val_cut_off
juglab · Aug 8, 2022 · eb09b9b · eb09b9b
2 parents 423d272 + ac87e86
commit eb09b9b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cryocare/internals/CryoCAREDataModule.py b/cryocare/internals/CryoCAREDataModule.py
@@ -212,8 +212,8 @@ def __compute_extraction_shapes__(self, even_path, odd_path, tilt_axis_index, sa
         assert even.data.shape[2] > 2 * sample_shape[2]
 
         val_cut_off = int(even.data.shape[tilt_axis_index] * validation_fraction)
-        if even.data.shape[tilt_axis_index] - val_cut_off < sample_shape[tilt_axis_index]:
-            val_cut_off = even.data.shape[tilt_axis_index] - sample_shape[tilt_axis_index]
+        if ((even.data.shape[tilt_axis_index] - val_cut_off) < sample_shape[tilt_axis_index]) or val_cut_off < sample_shape[tilt_axis_index]:
+            val_cut_off = even.data.shape[tilt_axis_index] - sample_shape[tilt_axis_index] - 1
 
         extraction_shape_train = [[0, even.data.shape[0]], [0, even.data.shape[1]], [0, even.data.shape[2]]]
         extraction_shape_val = [[0, even.data.shape[0]], [0, even.data.shape[1]], [0, even.data.shape[2]]]