From 87e079a3db44ff6badac7b697ca16c3dc72207e3 Mon Sep 17 00:00:00 2001
From: Emmanuel Moebel <emoebel@igrida-frontend.irisa.fr>
Date: Fri, 19 Nov 2021 17:25:16 +0100
Subject: [PATCH] uncommented a line in seg_single_block
---
deepfinder/inference.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/deepfinder/inference.py b/deepfinder/inference.py
index 535d2b5..d8624bd 100644
--- a/deepfinder/inference.py
+++ b/deepfinder/inference.py
@@ -130,7 +130,7 @@ 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, :, :, :, :]
--
GitLab