From 092c214103fd199ae9982651101069fa6c63338a Mon Sep 17 00:00:00 2001 From: deeppomf Date: Tue, 27 Feb 2018 11:34:14 -0500 Subject: [PATCH] decensor now uses args --- decensor.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/decensor.py b/decensor.py index 890ecdf..3526d4a 100644 --- a/decensor.py +++ b/decensor.py @@ -11,15 +11,13 @@ from model import Model from poisson_blend import blend from config import * -#size of input of local discrimnator. do not change this value. -LOCAL_SIZE = 64 BATCH_SIZE = 1 image_folder = 'decensor_input_images/' mask_color = [0, 255, 0] poisson_blending_enabled = False -def decensor(): +def decensor(args): x = tf.placeholder(tf.float32, [args.batch_size, args.image_size, args.image_size, args.input_channel_size]) mask = tf.placeholder(tf.float32, [args.batch_size, args.image_size, args.image_size, 1]) local_x = tf.placeholder(tf.float32, [args.batch_size, args.local_image_size, args.local_image_size, args.input_channel_size]) @@ -82,5 +80,5 @@ def get_mask(x_batch): return np.array(mask) if __name__ == '__main__': - decensor() + decensor(args)