Merge pull request #56 from bertsky/non-verbose

suppress Keras progress meter
2 years ago · ae45802f61
parent 39ef3fd7bb bb1484bd2b
commit ae45802f61
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]