From ccc05b8cef8b0a66964372dc3a2f51fc8709d362 Mon Sep 17 00:00:00 2001
From: emoebel <emmanuel.moebel@inria.fr>
Date: Thu, 15 Apr 2021 10:00:45 +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 4bbd5f3..755518f 100644
--- a/deepfinder/inference.py
+++ b/deepfinder/inference.py
@@ -131,7 +131,7 @@ class Segment(core.DeepFinder):
dim = dataArray.shape
dataArray = (dataArray[:] - np.mean(dataArray[:])) / np.std(dataArray[:]) # normalize
- dataArray = np.pad(dataArray, self.pcrop) # zeropad
+ dataArray = np.pad(dataArray, self.pcrop, mode='constant') # zeropad
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