From e6d5312dc1ec8c966fc8cc313d5cd2c563492969 Mon Sep 17 00:00:00 2001
From: emoebel <emmanuel.moebel@inria.fr>
Date: Thu, 15 Apr 2021 10:03:56 +0200
Subject: [PATCH] corrected small bug in segment.launch_single_block
---
deepfinder/inference.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/deepfinder/inference.py b/deepfinder/inference.py
index 755518f..9d500e3 100644
--- a/deepfinder/inference.py
+++ b/deepfinder/inference.py
@@ -129,9 +129,9 @@ class Segment(core.DeepFinder):
def launch_single_block(self, dataArray):
self.check_arguments(dataArray, self.P)
- dim = dataArray.shape
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]
pred = self.net.predict(dataArray, batch_size=1)
--
GitLab