mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-09 04:10:00 +02:00
pil2cv: reduce transparency to white
This commit is contained in:
parent
f11d0b0bf7
commit
88d64de67b
1 changed files with 4 additions and 0 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue