from PIL import Image image = Image.open('mask2.png') def toBlackWhiteMask(input_pil): if input_pil.mode != 'RGBA': input_pil = input_pil.convert('RGBA') new_image = Image.new('RGBA', input_pil.size, (255, 255, 255, 255)) input_data = input_pil.getdata() new_data = [] for item in input_data: if item[3] == 0: new_data.append((0, 0, 0, 255)) else: new_data.append((255, 255, 255, 255)) new_image.putdata(new_data) return new_image transform_image(image).save('mask2_nice.png')