Exemplo n.º 1
0
config.epoch = FLAGS.epoch
config.learning_rate = FLAGS.learning_rate  # optim learn rate
config.beta1 = FLAGS.beta1  # momentum
config.repeat_data = True
config.shuffle_data = True
config.buffer_size = 4
config.drop_remainder = True  # currently fails if false!
config.gener_iter = FLAGS.gener_iter  # times to update generator per discriminator update
config.noisy_inputs = False  # add some small noise to the input images
config.flip_inputs = False  # whether to flip the black white pixels

# i/o structures
config.log_dir = 'log'
config.out_dir = 'out'
config.samp_dir = 'samp'
config.chkp_dir = 'chkp'
config.paint_dir = 'paint'
config.post_dir = 'post'
config.run_dir = FLAGS.run_dir
if not config.run_dir:  # if the run dir was not given, make something up
    config.run_dir = rand_id()

# painting configurations
pconfig = Config()
pconfig.label = FLAGS.paint_label
pconfig.width = FLAGS.paint_width
pconfig.height = FLAGS.paint_height
pconfig.overlap = FLAGS.paint_overlap
pconfig.patcher = FLAGS.paint_patcher
pconfig.overlap_thresh = FLAGS.paint_overlap_thresh
pconfig.groundtruth = FLAGS.paint_groundtruth