From 10a0a44bc97cc8b62faa7c20f717de0b52dfcda0 Mon Sep 17 00:00:00 2001 From: Cameron Date: Sat, 8 Oct 2016 16:21:04 -0700 Subject: Fixed 'style_mask' error --- neural_style.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'neural_style.py') diff --git a/neural_style.py b/neural_style.py index 74badee..0db8403 100644 --- a/neural_style.py +++ b/neural_style.py @@ -528,7 +528,7 @@ def stylize(content_img, style_imgs, init_img, frame=None): net = build_vgg19(content_img) # style loss - if args.has_style_mask: + if args.style_mask: L_style = sum_masked_style_losses(sess, net, style_imgs) else: L_style = sum_style_losses(sess, net, style_imgs) -- cgit v1.2.3-70-g09d2