From 88d64de67b63fccc41d2d569927b8d790f7837d8 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Tue, 11 Oct 2022 07:20:38 +0000 Subject: [PATCH] pil2cv: reduce transparency to white --- sbb_binarize/ocrd_cli.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sbb_binarize/ocrd_cli.py b/sbb_binarize/ocrd_cli.py index 57438d3..9737bad 100644 --- a/sbb_binarize/ocrd_cli.py +++ b/sbb_binarize/ocrd_cli.py @@ -30,6 +30,10 @@ def cv2pil(img): def pil2cv(img): # from ocrd/workspace.py + if img.mode in ('LA', 'RGBA'): + newimg = Image.new(img.mode[:-1], img.size, 'white') + newimg.paste(img, mask=img.getchannel('A')) + img = newimg color_conversion = cv2.COLOR_GRAY2BGR if img.mode in ('1', 'L') else cv2.COLOR_RGB2BGR pil_as_np_array = np.array(img).astype('uint8') if img.mode == '1' else np.array(img) return cv2.cvtColor(pil_as_np_array, color_conversion)