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

added comment about batch_size

Browse Source
This commit is contained in:
deeppomf 2018-02-27 23:15:57 -05:00
parent 3cdf350e18
commit 018c86c959
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
decensor.py
View File

@ -11,6 +11,7 @@ from model import Model
from poisson_blend import blend from poisson_blend import blend
from config import * from config import *
#TODO: allow variable batch sizes when decensoring. changing BATCH_SIZE will likely result in crashing
BATCH_SIZE = 1 BATCH_SIZE = 1
mask_color = [0, 255, 0] mask_color = [0, 255, 0]
@ -81,5 +82,4 @@ def get_mask(x_batch):
if __name__ == '__main__': if __name__ == '__main__':
if not os.path.exists(args.decensor_output_path): if not os.path.exists(args.decensor_output_path):
os.makedirs(args.decensor_output_path) os.makedirs(args.decensor_output_path)
decensor(args) decensor(args)