Merge pull request #56 from bertsky/non-verbose

suppress Keras progress meter
This commit is contained in:
Clemens Neudecker 2023-03-29 21:56:42 +02:00 committed by GitHub
commit ae45802f61
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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]