pil2cv: reduce transparency to white

pull/52/head
Robert Sachunsky 2 years ago
parent f11d0b0bf7
commit 88d64de67b

@ -30,6 +30,10 @@ def cv2pil(img):
def pil2cv(img): def pil2cv(img):
# from ocrd/workspace.py # 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 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) 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) return cv2.cvtColor(pil_as_np_array, color_conversion)

Loading…
Cancel
Save