File size: 551 Bytes
2067800 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from PIL import Image
import os
def augment(img_path):
ROOT = "./images/VLAT"
NEW_ROOT = "./changed"
img = Image.open(f"{ROOT}/{img_path}").convert("RGBA")
# Create a white background
white_bg = Image.new("RGB", img.size, (255, 255, 255))
white_bg.paste(img, mask=img.split()[3]) # Paste using alpha channel as mask
# Save the result
white_bg.save(f'{NEW_ROOT}/{img_path}') # Or save as .png if you prefer
if __name__ == "__main__":
ROOT = "./images/VLAT"
for img in os.listdir(ROOT):
augment(img) |