diff options
| author | Cameron <cysmith1010@gmail.com> | 2016-10-08 16:21:04 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-10-08 16:21:04 -0700 |
| commit | 10a0a44bc97cc8b62faa7c20f717de0b52dfcda0 (patch) | |
| tree | ac1c90ac1c6d90e88f7b0c88346282c78255f810 /neural_style.py | |
| parent | 9af705f1786cdc157cf2bfb09aa8f95dab3f1122 (diff) | |
Fixed 'style_mask' error
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) |