Remove concatenate as now the prediction is inserted into an array di…

…rectly
BiaPyX · Mar 6, 2024 · 20fc2f7 · 20fc2f7
1 parent 79982d3
commit 20fc2f7
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/biapy/engine/denoising.py b/biapy/engine/denoising.py
@@ -152,7 +152,6 @@ def process_sample(self, norm):
         del self._X, p
 
         # Reconstruct the predictions
-        pred = np.concatenate(pred)
         if original_data_shape[1:-1] != self.cfg.DATA.PATCH_SIZE[:-1]:
             if self.cfg.PROBLEM.NDIM == '3D': original_data_shape = original_data_shape[1:]
             f_name = merge_data_with_overlap if self.cfg.PROBLEM.NDIM == '2D' else merge_3D_data_with_overlap

diff --git a/biapy/engine/image_to_image.py b/biapy/engine/image_to_image.py
@@ -156,7 +156,6 @@ def process_sample(self, norm):
         del self._X, p
 
         # Reconstruct the predictions
-        pred = np.concatenate(pred)
         if original_data_shape[1:-1] != self.cfg.DATA.PATCH_SIZE[:-1]:
             if self.cfg.PROBLEM.NDIM == '3D': original_data_shape = original_data_shape[1:]
             f_name = merge_data_with_overlap if self.cfg.PROBLEM.NDIM == '2D' else merge_3D_data_with_overlap