generate.py 文件源码

def generate(p, ics, gcs, *args):
    from scipy import misc
    import subprocess
    import datetime
    inits = p.generate(np.array(ics),*args)
    goals = p.generate(np.array(gcs),*args)
    for noise_fn,output_dir in zip(noise_fns,output_dirs):
        inits = noise_fn(inits)
        goals = noise_fn(goals)
        for i,init in enumerate(inits):
            for j,goal in enumerate(goals):
                d = "{}/{}/{:03d}-{:03d}-{:03d}".format(output_dir,p.__name__,steps,i,j)
                try:
                    subprocess.call(["mv",d,d+"_old_"+datetime.datetime.today().isoformat()])
                except:
                    pass
                os.makedirs(d)
                print(d)
                misc.imsave(os.path.join(d,"init.png"),init)
                misc.imsave(os.path.join(d,"goal.png"),goal)

################################################################