search

ProofByConstruction / texture-networks

Implementation of Texture Networks in Tensorflow
27 stars 9 forks source link

speed #6

Open benslaney opened 8 years ago

benslaney commented 8 years ago

i got this working, and it did work okay, but it's actually extremely slow. about an hour to generate an image (the same as the slow artistic style networks). would you have any idea why?

hoqqanen commented 8 years ago

Hey Ben, it's very slow to run this on a CPU. For decent performance you'll want to use a GPU. Running on AWS with a tensorflow+GPU AMI will get you a decent speedup (a couple minutes per image).

The fast style transfer via texture networks requires additional training and I haven't taken the time to make them work yet. For a torch implementation see https://github.com/DmitryUlyanov/texture_nets (it will take a while to train the nets -- a long long while without a very fast graphics card).