NeuralNeighborStyleTransfer
NeuralNeighborStyleTransfer copied to clipboard
nkolkin13
Optimization based style transfer
We are looking to run this model and reduce the processing time. Is there any way to tune hyperparameters and/or parallelise inference so that we can leverage more compute?
I'm trying to install NNST on an ARM Ubuntu 22.04 virtual machine. Seems like Torch doesn't work on that setup? ``` python styleTransfer.py --content_path inputs/content/C1.png --style_path inputs/style/S4.jpg --output_path ./output.jpg --cpu...
For instance, with a 850x756 content image: ``` File "X:\Avirtual\NeuralNeighborStyleTransfer-main\styleTransfer.py", line 70, in output = produce_stylization(content_im_orig, style_im_orig, phi, File "X:\Avirtual\NeuralNeighborStyleTransfer-main\utils\stylize.py", line 122, in produce_stylization s_pyr = optimize_output_im(s_pyr, c_pyr, content_im, style_im_tmp,...
Why do I get this error using 3090 with 24GB of memory RuntimeError: CUDA error: out of memory Thank you for your answer!!!
Someone asked me if i could turn this into a colab for them, this is just simple demo to take in single images and not batches or videos https://colab.research.google.com/drive/1Osok7GPHe0uHMpEn2L6xqqdRUEZhGj7X?usp=sharing _Sent...
Hi there 👋🏻 We noticed some issues with logging for [your model on replicate](https://replicate.com/nkolkin13/neuralneighborstyletransfer), and pushed a new version built with a more recent version of Cog to address them....
Hello, First thank you for your work and sharing it ! I tinkered a bit with you project yesterday and deployed it to Gradio, I can do a PR and...
Is it possible to retain the colors of the content image? The color preservation does only work sometimes.