From 6fe8a3228e3170600b68d9b19e3b8f07e51222fe Mon Sep 17 00:00:00 2001
From: xAlpharax <42233094+xAlpharax@users.noreply.github.com>
Date: Sun, 26 Feb 2023 02:50:33 +0200
Subject: massive changes in the pipeline and ease of use

---
 neuralart.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'neuralart.py')

diff --git a/neuralart.py b/neuralart.py
index 4b29d21..ebbbb94 100644
--- a/neuralart.py
+++ b/neuralart.py
@@ -147,7 +147,7 @@ prep = transforms.Compose([transforms.Resize(img_size),
                            ])
 
 # POST-PROCESSING A
-postpa = transforms.Compose([transforms.Lambda(lambda x: x.mul_(1./255)),
+postpa = transforms.Compose([transforms.Lambda(lambda x: x.mul_(1./255)), # REVERT EVERYTHING DONE IN THE PRE-PROCESSING STEP
                             transforms.Normalize(mean = [-0.40760392, -0.45795686, -0.48501961], std = [1, 1, 1]),
                             transforms.Lambda(lambda x: x[torch.LongTensor([2,1,0])])
                             ])
@@ -234,7 +234,7 @@ targets = style_targets + content_targets
 import numpy as np
 from tqdm import tqdm
 
-vis_factor = 3
+vis_factor = 5
 max_iter = 600 * vis_factor
 show_iter = 1 * vis_factor
 
-- 
cgit v1.2.3