synthesize_image_for_train.py

from PIL import Image, ImageDraw, ImageFilter
from pathlib import Path
import random


def getImage(path):
    img = resizeImage(Image.open(path))
    return img


def resizeImage(image):
    return image.resize((640, 640))


def getAllFilePath(dirPath):
    p = Path(dirPath)
    flist = list(p.glob('*.jpg'))
    all_path = []
    for f in flist:
        all_path.append(str(f.resolve()))
    return all_path


def getTargetImages(dirPath):
    p = Path(dirPath)
    flist = list(p.glob('*.png'))
    imgs = []
    for i, f in enumerate(flist):
        # f.rename('squirrel_' + str(i))
        img = Image.open(f)
        imgs.append(img.resize((400, 400)))
    return imgs


def run():
    # get target obj img
    objs = getTargetImages('./squirrel_base')

    # list files
    files = getAllFilePath('./challenge2018')
    for i, fpath in enumerate(files):
        if i > 1000:
            return
        bg_img = getImage(fpath)
        # add alpha channel
        bg_img.putalpha(255)

        # choise target img random
        obj = objs[random.randrange(0, len(objs) - 1)]

        # paste obj to bg image
        bg_img.paste(obj, (120, 120), obj)
        # jpg_img = bg_img.convert('RGB')
        bg_img.save('treat/img_' + str(i) + '.png')


run()