Open xsy123xsy opened 3 years ago
I haven't tried ONNX, but the model converted to pytorch and the example of image conversion(tiled rendering) with pytorch can be found in https://github.com/nagadomi/nunif
(You need to install git-lfs
to pull the pretrained model files)
model parameter/state conversion script: https://github.com/nagadomi/nunif/blob/master/nunif/cli/import_waifu2x_models.py
Also, waifu2x-caffe has a model converted to caffe model.
Thanks for help, I will try to transform the model from pytorch to onnx. By the way, to do the transformation, can I use python script like this: https://github.com/HollowMan6/SRMD-Pytorch/blob/main/pytorch2onnx.py
------------------ 原始邮件 ------------------ 发件人: "nagadomi/waifu2x" <notifications@github.com>; 发送时间: 2021年2月6日(星期六) 上午6:02 收件人: "nagadomi/waifu2x"<waifu2x@noreply.github.com>; 抄送: "xsy_xsy"<366366235@qq.com>;"Author"<author@noreply.github.com>; 主题: Re: [nagadomi/waifu2x] transform the model to onnx (#382)
I haven't tried ONNX, but the model converted to pytorch and the example of image conversion(tiled rendering) with pytorch can be found in https://github.com/nagadomi/nunif (You need to install git-lfs to pull the pretrained model files)
model parameter/state conversion script: https://github.com/nagadomi/nunif/blob/master/nunif/cli/import_waifu2x_models.py
— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.
I am trying to test the model in a Real Time video process engine, before that I need to transform the model into onnx Can I use a python script like this: