From 42af784ae5459142fc14e023e5bb726e800ac4a9 Mon Sep 17 00:00:00 2001 From: balajicoding <164991652+balajicoding@users.noreply.github.com> Date: Mon, 21 Apr 2025 20:23:05 +0530 Subject: [PATCH] Update neural_style_transfer.py --- neural_style_transfer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/neural_style_transfer.py b/neural_style_transfer.py index bfc70e6..542f625 100644 --- a/neural_style_transfer.py +++ b/neural_style_transfer.py @@ -142,7 +142,8 @@ def closure(): parser = argparse.ArgumentParser() parser.add_argument("--content_img_name", type=str, help="content image name", default='figures.jpg') parser.add_argument("--style_img_name", type=str, help="style image name", default='vg_starry_night.jpg') - parser.add_argument("--height", type=int, help="height of content and style images", default=400) + parser.add_argument('--height', type=int, default=256, help='Height of the output image.') + parser.add_argument("--content_weight", type=float, help="weight factor for content loss", default=1e5) parser.add_argument("--style_weight", type=float, help="weight factor for style loss", default=3e4)