Fix transpose

FZJ-INM1-BDA · Jul 9, 2021 · 019702c · 019702c
1 parent 29e6721
commit 019702c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/celldetection/data/misc.py b/celldetection/data/misc.py
@@ -16,7 +16,7 @@ def transpose_spatial(inputs: np.ndarray, inputs_channels_last=True, spatial_dim
     else:
         # e.g. (0, 2, 3, 1)
         b = list(range(inputs.ndim - spatial_dims, inputs.ndim))  # spatial dims
-        c = list(range(has_batch, inputs.ndim - spatial_dims - (1 - has_batch)))  # n channels
+        c = list(range(has_batch, inputs.ndim - spatial_dims))  # n channels
     return np.transpose(inputs, a + b + c)