qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-06-09 04:10:00 +02:00
Code Issues Releases Wiki Activity

pil2cv: reduce transparency to white

Browse source
This commit is contained in:
Robert Sachunsky 2022-10-11 07:20:38 +00:00
parent f11d0b0bf7
commit 88d64de67b
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
sbb_binarize/ocrd_cli.py
View file

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