Will it have a pytorch version?

luanfujun / deep-painterly-harmonization

Code and data for paper "Deep Painterly Harmonization": https://arxiv.org/abs/1804.03189

6.07k stars 626 forks source link

Will it have a pytorch version? #7

Closed a1324321 closed 6 years ago

a1324321 commented 6 years ago

Will it have a pytorch version? I would very appreciate it.

luanfujun commented 6 years ago

Currently no... 😂

GuohongLi commented 6 years ago

Maybe, a tensorflow or pytorch version will be welcomed. Thx

a1324321 commented 6 years ago

😂

ProGamerGov commented 6 years ago

Deep Painterly Harmonization is based on Neural-Style, which was written in Lua. If you wanted to convert it to PyTorch, I converted Justin Johnson's Neural-Style to PyTorch a while back here: https://github.com/ProGamerGov/neural-style-pt. I tried to mimic the original Lua code as much as possible, so it should be pretty easy to rebuilt Deep Painterly Harmonization off of it if you know both Python and Lua.