qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-06-30 22:49:54 +02:00
Code Issues Releases Wiki Activity

silentium!

Browse source
This commit is contained in:
Robert Sachunsky 2023-02-11 20:00:27 +00:00
parent 39ef3fd7bb
commit bb1484bd2b
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
sbb_binarize/sbb_binarize.py
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]