search

ProGamerGov / neural-style-pt

PyTorch implementation of neural style transfer algorithm
MIT License
833 stars 178 forks source link

Is a batch process possible? #92

Closed jamahun closed 2 years ago

jamahun commented 2 years ago

I'm wondering if anyone has successfully created a batch process for this script so that instead of specifying a single item for the style image you could specify a list or a whole directory? The intention would be that once it finished stylising item 1 in the list it automatically starts processing item 2, item 3 etc etc.. I know this would mean creating some kind of loop but I don't have the technical knowhow to set it up. I image it would have to be some kind of shell script similar to the starry_stanford.sh file in the scripts folder

jamahun commented 2 years ago

So I ended up getting a loop to work with the pip installation technique and by re-formatting the code in python. The only thing I can't figure out is how to make make the output parameter work with a variable + ext

Any suggestion greatly appreciated.

https://colab.research.google.com/drive/1uKrDRfunX9ZdbPQ3_b2cfwKxmh43anTG?usp=sharing