diff options
| author | xAlpharax <42233094+xAlpharax@users.noreply.github.com> | 2023-02-26 02:50:33 +0200 |
|---|---|---|
| committer | xAlpharax <42233094+xAlpharax@users.noreply.github.com> | 2023-02-26 02:50:33 +0200 |
| commit | 6fe8a3228e3170600b68d9b19e3b8f07e51222fe (patch) | |
| tree | a648ce29c376aa32f3ace8f18b967ca3d5ea4b55 /neuralart.py | |
| parent | 4f51d2b619b5cd359ad853494cb826aa5e29495e (diff) | |
massive changes in the pipeline and ease of use
Diffstat (limited to 'neuralart.py')
| -rw-r--r-- | neuralart.py | 4 |
1 files changed, 2 insertions, 2 deletions
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 |