generate_esposalles_images.py

from docemul.generator import generate
from docemul.backgroundmodel import RealBackGound
from docemul.augment import data_augment

RealBackGound.dirname = 'Data/Esposalles/Background'

realSampler = RealBackGound.load_examples()

def run(dir, num=1, size=None, sampler=realSampler, greyscale=True,type='IMG',model='data/Eesposalles/esposalles.xml',seed=1):
    generate(dir, num=num, size=size,sampler=sampler, greyscale=greyscale,type=type,model=model,seed=seed)

#generate dataset with background
run('GENERATED/Esposalles/test',size=(366*4,256*4),num=1)

#data_augmentation
data_augment('GENERATED/Esposalles/test/gt.csv', 'GENERATED/Esposalles/test_augmented',f_output = 'gt_augment.csv',resize=None,rotate=2, rotate_time=1, noise=1)