From 63de829bd1a805cf429880a59178330e0faeb263 Mon Sep 17 00:00:00 2001 From: ryletko Date: Sun, 17 Jan 2021 21:26:05 +0500 Subject: [PATCH 1/2] added style scale feature --- neural_style_transfer.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/neural_style_transfer.py b/neural_style_transfer.py index bfc70e6..c7ee25e 100644 --- a/neural_style_transfer.py +++ b/neural_style_transfer.py @@ -26,7 +26,7 @@ def build_loss(neural_net, optimizing_img, target_representations, content_featu tv_loss = utils.total_variation(optimizing_img) - total_loss = config['content_weight'] * content_loss + config['style_weight'] * style_loss + config['tv_weight'] * tv_loss + total_loss = config['content_weight'] * content_loss + config['style_weight'] * (style_loss) + config['tv_weight'] * tv_loss return total_loss, content_loss, style_loss, tv_loss @@ -57,7 +57,7 @@ def neural_style_transfer(config): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") content_img = utils.prepare_img(content_img_path, config['height'], device) - style_img = utils.prepare_img(style_img_path, config['height'], device) + style_img = utils.prepare_img(style_img_path, int(config['height'] * config['style_scale']), device) if config['init_method'] == 'random': # white_noise_img = np.random.uniform(-90., 90., content_img.shape).astype(np.float32) @@ -152,6 +152,8 @@ def closure(): parser.add_argument("--model", type=str, choices=['vgg16', 'vgg19'], default='vgg19') parser.add_argument("--init_method", type=str, choices=['random', 'content', 'style'], default='content') parser.add_argument("--saving_freq", type=int, help="saving frequency for intermediate images (-1 means only final)", default=-1) + parser.add_argument("--style_scale", type=float, help="style scale (1 means no scale)", default=1) + args = parser.parse_args() # some values of weights that worked for figures.jpg, vg_starry_night.jpg (starting point for finding good images) From 6361aa6a7cd6e8e6ca20408dfa94ac24bc6bd71d Mon Sep 17 00:00:00 2001 From: ryletko Date: Sun, 17 Jan 2021 21:34:15 +0500 Subject: [PATCH 2/2] removed parentheses --- neural_style_transfer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_style_transfer.py b/neural_style_transfer.py index c7ee25e..bab71f8 100644 --- a/neural_style_transfer.py +++ b/neural_style_transfer.py @@ -26,7 +26,7 @@ def build_loss(neural_net, optimizing_img, target_representations, content_featu tv_loss = utils.total_variation(optimizing_img) - total_loss = config['content_weight'] * content_loss + config['style_weight'] * (style_loss) + config['tv_weight'] * tv_loss + total_loss = config['content_weight'] * content_loss + config['style_weight'] * style_loss + config['tv_weight'] * tv_loss return total_loss, content_loss, style_loss, tv_loss