This commit is contained in:
deeppomf 2018-03-14 21:04:45 -04:00
parent 1f4f847008
commit 85c9886f41

View File

@ -14,7 +14,7 @@ from config import *
#TODO: allow variable batch sizes when decensoring. changing BATCH_SIZE will likely result in crashing
BATCH_SIZE = 1
mask_color = [args.mask_color_red, args.mask_color_green, arg.mask_color_blue]
mask_color = [args.mask_color_red, args.mask_color_green, args.mask_color_blue]
poisson_blending_enabled = False
def decensor(args):