search

manuelruder / artistic-videos

Torch implementation for the paper "Artistic style transfer for videos"
Other
1.75k stars 252 forks source link

deepmatching GPU code #41

Open filmo opened 7 years ago

filmo commented 7 years ago

As a side question, has anybody managed to get the deepmatching GPU code to compile on Ubuntu? It seems to have been written against some version of caffe/protobuf that no longer seem compatible once the SWIG code is compiled. Or perhaps I'm just really doing something wrong. I've beaten my head against it for a couple days now with zero progress.

I'm currently using the CPU static provided by the original author's site which works fine, but would love to be able to speed up the matching process.

WIStudent commented 7 years ago

Hi. I am currently using DeepMatching for my master thesis and just managed to get the GPU version running on Ubuntu. Here is how I did it https://gist.github.com/WIStudent/08072e8dd41487d2dde7ce75eec3dcbf

filmo commented 7 years ago

Awesome. Thanks. I'll give it a try later today. What performance boost did the GPU code give over the Numpy based CPU version ?

WIStudent commented 7 years ago

I haven't done any performance tests yet, I just got it running yesterday.