qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-10-31 01:24:14 +01:00
Code Issues Releases Wiki Activity

Adding a bit more documentation

Browse source
This commit is contained in:
Alexander Pacha 2022-08-30 11:54:04 +02:00
parent 9db2bcc8b0
commit 7edf34a815
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
sbb_binarize/sbb_binarize.py
View file

@ -67,9 +67,13 @@ class SbbBinarizer:
gc.collect()
tf.keras.backend.clear_session()
binary_patches = np.invert(np.argmax(predicted_patches, axis=3).astype(bool)).astype(np.uint8) * 255
# The result is a white-on-black image that needs to be inverted to be displayed as black-on-white image
# We do this by converting the binary values to a boolean numpy-array and then inverting the values
black_on_white_patches = np.invert(np.argmax(predicted_patches, axis=3).astype(bool))
# cv2 can't export a boolean numpy array into a black-and-white PNG image, so we have to convert it to uint8 (grayscale) values
grayscale_patches = black_on_white_patches.astype(np.uint8) * 255
full_image_with_padding = self._patches_to_image(
binary_patches,
grayscale_patches,
padded_image_height,
padded_image_width,
self.model_height,