mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2024-11-29 05:10:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

decensor now uses args

Browse Source
This commit is contained in:
deeppomf 2018-02-27 11:34:14 -05:00
parent 967f6d22dd
commit 092c214103
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
decensor.py
View File

@ -11,15 +11,13 @@ from model import Model
from poisson_blend import blend from poisson_blend import blend
from config import * from config import *
#size of input of local discrimnator. do not change this value.
LOCAL_SIZE = 64
BATCH_SIZE = 1 BATCH_SIZE = 1
image_folder = 'decensor_input_images/' image_folder = 'decensor_input_images/'
mask_color = [0, 255, 0] mask_color = [0, 255, 0]
poisson_blending_enabled = False 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]) 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]) 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]) 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) return np.array(mask)
if __name__ == '__main__': if __name__ == '__main__':
decensor() decensor(args)