From 1d6bafb6c85eefa308fc607112533aa2dd9b5feb Mon Sep 17 00:00:00 2001
From: emoebel <emmanuel.moebel@inria.fr>
Date: Thu, 15 Apr 2021 10:06:34 +0200
Subject: [PATCH] corrected small bug in segment.launch_single_block
---
deepfinder/inference.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/deepfinder/inference.py b/deepfinder/inference.py
index 9d500e3..a63bc72 100644
--- a/deepfinder/inference.py
+++ b/deepfinder/inference.py
@@ -132,11 +132,11 @@ class Segment(core.DeepFinder):
dataArray = (dataArray[:] - np.mean(dataArray[:])) / np.std(dataArray[:]) # normalize
dataArray = np.pad(dataArray, self.pcrop, mode='constant') # zeropad
dim = dataArray.shape
- dataArray = np.reshape(dataArray, (1, dim[0], dim[1], dim[2], 1)) # reshape for keras [batch,x,y,z,channel]
+ #dataArray = np.reshape(dataArray, (1, dim[0], dim[1], dim[2], 1)) # reshape for keras [batch,x,y,z,channel]
pred = self.net.predict(dataArray, batch_size=1)
predArray = pred[0, :, :, :, :]
- predArray = predArray[self.pcrop:-self.pcrop, self.pcrop:-self.pcrop, self.pcrop:-self.pcrop, :] # unpad
+ #predArray = predArray[self.pcrop:-self.pcrop, self.pcrop:-self.pcrop, self.pcrop:-self.pcrop, :] # unpad
return predArray
--
GitLab