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