diff options
Diffstat (limited to 'neural_style.py')
| -rw-r--r-- | neural_style.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |