diff --git a/deepfinder/inference.py b/deepfinder/inference.py index 4bbd5f38350daf0aa344bc04b9e7434ab1d63cec..755518f13476035123a48a8592cd66222eb6615b 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)