def main(_):
assert(os.path.exists(FLAGS.checkpointDir))
# Deal with input size default values
if not FLAGS.input_width:
FLAGS.input_width = FLAGS.input_height
def get_default_gpu_session(fraction=0.8):
if fraction > 1:
fraction = 0.8
config = tf.ConfigProto(allow_soft_placement=True)
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = fraction
return tf.Session(config=config)
with get_default_gpu_session(FLAGS.gpu_utilization) as sess:
dcgan_instance = DCGAN(
sess,
input_width=FLAGS.input_width,
input_height=FLAGS.input_height,
c_dim=FLAGS.c_dim,
dataset_name=FLAGS.dataset,
input_fname_pattern=FLAGS.input_file_extension,
is_crop=FLAGS.is_crop,
checkpoint_dir=FLAGS.checkpointDir,
lambda_val=FLAGS.lambda_val,
# Xx-GAN
dataset_name2=FLAGS.dataset2
)
dcgan_instance.inpaint(FLAGS)
