summaryrefslogtreecommitdiff
path: root/neuralart.py
diff options
context:
space:
mode:
authorxAlpharax <42233094+xAlpharax@users.noreply.github.com>2023-12-29 22:32:28 +0200
committerxAlpharax <42233094+xAlpharax@users.noreply.github.com>2023-12-29 22:32:28 +0200
commit447b6c6c921f912e5323af2ed18160d63931533f (patch)
tree0f6709a0f70e555c91df9dc7e30a8f4763b70ee8 /neuralart.py
parent44bcbe8825cec9245373e7e807a615d4441b6287 (diff)
Great changes to the README, featuring full showcase of video outputs.
Main scripts have seen a couple QOS changes as well Changes to be committed: modified: .gitignore modified: README.md modified: neuralart.py modified: renderer.py modified: stylize.sh
Diffstat (limited to 'neuralart.py')
-rw-r--r--neuralart.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/neuralart.py b/neuralart.py
index fadaf12..9dc91e5 100644
--- a/neuralart.py
+++ b/neuralart.py
@@ -29,11 +29,11 @@ image_path = '' # by default use neural-art as relative dir
### userland testing for multiple instances, a big nono currently
-n_instances = os.popen('ps aux | grep "python neuralart.py" | wc -l').read()
+n_instances = os.popen('ps aux | grep "python neuralart.py" | wc -l').read() # TODO: add windows commands for platform compatibility :p for the 3 people who need this warning
if int(n_instances) > 3: print("Woah, running 2 or more instances of neural-art at the same time?\nThis is an experimental feature as of now... try it later :3")
### check if there are any weights to use, if not, download the default provided ones
-if int(os.popen('ls -l weights | wc -l').read()) == 1: os.system(f'curl https://files.catbox.moe/wcao20.pth --output {model_path}')
+if int(os.popen('ls -l weights | wc -l').read()) == 1: os.system(f'curl https://files.catbox.moe/wcao20.pth --output {model_path}') # TODO: win commands here as well
### Defining neural architecture
@@ -244,7 +244,7 @@ targets = style_targets + content_targets
import numpy as np
from tqdm import tqdm
-vis_factor = 10
+vis_factor = 10 # every 10 iterations, a frame snapshot is saved, we use this coeficient to scale
max_iter = 600 * vis_factor
show_iter = 1 * vis_factor