silentium!

2 years ago · bb1484bd2b
parent 39ef3fd7bb
commit bb1484bd2b
1 changed files with 2 additions and 1 deletions
--- a/sbb_binarize/sbb_binarize.py
+++ b/sbb_binarize/sbb_binarize.py
@ -154,7 +154,8 @@ class SbbBinarizer:
                img_patch = img[index_y_d:index_y_u, index_x_d:index_x_u, :]
-                label_p_pred = model.predict(img_patch.reshape(1, img_patch.shape[0], img_patch.shape[1], img_patch.shape[2]))
+                label_p_pred = model.predict(img_patch.reshape(1, img_patch.shape[0], img_patch.shape[1], img_patch.shape[2]),
                                             verbose=0)
                seg = np.argmax(label_p_pred, axis=3)[0]