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

poisson blending variable

Browse Source
This commit is contained in:
deep pomf 2018-02-12 12:21:10 -05:00
parent 4ed26b5d06
commit 181e13ef68
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
decensor.py
View File

@ -15,8 +15,10 @@ HOLE_MIN = 24
HOLE_MAX = 48 HOLE_MAX = 48
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
def decensor(): def decensor():
x = tf.placeholder(tf.float32, [BATCH_SIZE, IMAGE_SIZE, IMAGE_SIZE, 3]) x = tf.placeholder(tf.float32, [BATCH_SIZE, IMAGE_SIZE, IMAGE_SIZE, 3])
@ -60,8 +62,8 @@ def decensor():
img = np.array((img + 1) * 127.5, dtype=np.uint8) img = np.array((img + 1) * 127.5, dtype=np.uint8)
original = x_batch[i] original = x_batch[i]
original = np.array((original + 1) * 127.5, dtype=np.uint8) original = np.array((original + 1) * 127.5, dtype=np.uint8)
#uncomment to enable poisson_blending, but poisson_blending doesn't seem to help if (poisson_blending_enabled):
#img = blend(original, img, mask_batch[0,:,:,0]) img = blend(original, img, mask_batch[0,:,:,0])
output = Image.fromarray(img.astype('uint8'), 'RGB') output = Image.fromarray(img.astype('uint8'), 'RGB')
dst = './decensor_output_images/{}.png'.format("{0:06d}".format(cnt)) dst = './decensor_output_images/{}.png'.format("{0:06d}".format(cnt))
output.save(dst) output.save(dst)