nagadomi / waifu2x

Image Super-Resolution for Anime-Style Art
http://waifu2x.udp.jp/
MIT License
27.56k stars 2.71k forks source link

Possible to port waifu2x models to other languages? #224

Open ghost opened 6 years ago

ghost commented 6 years ago

Hi, firstly I just want to say that this is an amazing project! I was wondering if the default model trained in this repo can be adapted for use in other languages. I ask this out of curiosity and also because I am interested in making a Python version of waifu2x. From my understanding, it seems that Torch is based on the TensorFlow framework which is available for numerous other languages.

Any tips on how I can start, or is this just not possible?

nagadomi commented 6 years ago

waifu2x's trained models are stored in the json file under the model directory. The definition of CNN is in https://github.com/nagadomi/waifu2x/blob/ead75a2c9cdae2da7c4e168250e76d18bdb09d91/lib/srcnn.lua#L141-L166

chungexcy's work will be helpful. https://github.com/chungexcy/waifu2x-new

Many of the derivation software of waifu2x also uses the json file in this repository. https://github.com/search?utf8=✓&q=waifu2x&type=

ghost commented 6 years ago

Thank you!

2ji3150 commented 6 years ago

Hope someday there is an ONNX model which can be use in the windows ML .