import os
from PIL import Image

root = '/home/mly/data/datasets/text_recognition/finetune/src/'
img_root = os.path.join(root, 'img')
mask_root = os.path.join(root, 'mask')
crop_root = os.path.join(root, 'crop')
blend_root = os.path.join(root, 'blend')

img_list = os.listdir(img_root)
if not os.path.exists(crop_root):
    os.mkdir(crop_root)

for name in img_list:
    img = Image.open(os.path.join(img_root, name)).convert('RGB')
    mask = Image.open(os.path.join(mask_root, name))

    crop = Image.new('RGB', img.size)
    crop.paste(img, mask=mask)