From d319517a87726b3b3e48231da641c44e1318c2bb Mon Sep 17 00:00:00 2001 From: "P.J. Finlay" Date: Wed, 6 Jan 2021 18:07:20 -0600 Subject: [PATCH 1/5] Added requirements.txt --- requirements.txt | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..4e54d2c --- /dev/null +++ b/requirements.txt @@ -0,0 +1,5 @@ +matplotlib +torch +torchvision +numpy +opencv-python From 4d8e8bcdd9648c26fc7019a678773969dfc6d095 Mon Sep 17 00:00:00 2001 From: "P.J. Finlay" Date: Fri, 15 Jan 2021 20:23:26 -0600 Subject: [PATCH 2/5] Changed default to maintaining current image size --- 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 bfc70e6..39ba47b 100644 --- a/neural_style_transfer.py +++ b/neural_style_transfer.py @@ -142,7 +142,7 @@ 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, help="height of content and style images", default=None) 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) From ca412b30cf67183848833300e0fa6fac11520fb6 Mon Sep 17 00:00:00 2001 From: "P.J. Finlay" Date: Sat, 16 Jan 2021 17:56:34 +0000 Subject: [PATCH 3/5] Added script to automate style transfer --- scripts/style_transfer_all.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100755 scripts/style_transfer_all.py diff --git a/scripts/style_transfer_all.py b/scripts/style_transfer_all.py new file mode 100755 index 0000000..cf8015e --- /dev/null +++ b/scripts/style_transfer_all.py @@ -0,0 +1,35 @@ +#!/usr/bin/env python3 + +# Does neural style transfer for each content image and style image pair +# Run from root of project + +import subprocess +from pathlib import Path + +DEBUG=True + +def run_command(args): + if DEBUG: + print(args) + else: + subprocess.run(args) + +run_command(['touch', 'test']) + +data_dir = Path('data') +content_dir = data_dir / 'content-images' +style_dir = data_dir / 'style-images' + +content_images = [f.name for f in list(content_dir.iterdir())] +style_images = [f.name for f in list(style_dir.iterdir())] + +i = 0 +for content_image in content_images: + for style_image in style_images: + run_command(['python3', 'neural_style_transfer.py', + '--content_img_name', content_image, + '--style_img_name', style_image, + '--height', 720]) + i += 1 + print('Finished image ' + str(i)) + From cabe2a7a491c1c23fea67375b0ee551d674e052f Mon Sep 17 00:00:00 2001 From: "P.J. Finlay" Date: Sat, 16 Jan 2021 18:12:30 +0000 Subject: [PATCH 4/5] Fixed bug in previous commit --- scripts/style_transfer_all.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/style_transfer_all.py b/scripts/style_transfer_all.py index cf8015e..0f789d5 100755 --- a/scripts/style_transfer_all.py +++ b/scripts/style_transfer_all.py @@ -6,7 +6,7 @@ import subprocess from pathlib import Path -DEBUG=True +DEBUG=False def run_command(args): if DEBUG: @@ -29,7 +29,7 @@ def run_command(args): run_command(['python3', 'neural_style_transfer.py', '--content_img_name', content_image, '--style_img_name', style_image, - '--height', 720]) + '--height', '720']) i += 1 print('Finished image ' + str(i)) From dfd11769b06363ef8a27c9f7b0dce393fffcf9e2 Mon Sep 17 00:00:00 2001 From: "P.J. Finlay" Date: Sat, 30 Jan 2021 13:49:22 +0000 Subject: [PATCH 5/5] Changed default height to 720 --- 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 39ba47b..d4c05b9 100644 --- a/neural_style_transfer.py +++ b/neural_style_transfer.py @@ -142,7 +142,7 @@ 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=None) + parser.add_argument("--height", type=int, help="height of content and style images", default=720) 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)