怎麽能在Cupscale GUI裏調用Real-ESRGAN的pytorch AI網絡？

xinntao / Real-ESRGAN

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

BSD 3-Clause "New" or "Revised" License

27.64k stars 3.47k forks source link

Open Battlecraft369 opened 2 years ago

Battlecraft369 commented 2 years ago

現在有個Cupscale GUI，目前沒辦法使用AI network：Real-ESRGAN（Pytorch）。只能用Real-ESRGAN（ncnnn）網絡。能指點一下把源碼的什麽部分拷貝出來或者怎麽搞一下，能讓這個圖形化界面能使用到Real-ESRGAN（Pytorch）網絡嗎？

個人是接觸這個的新手，對這個深度學習模型很感興趣，能加一下您的微信聯係方式，便於請教一些具體問題不？

Battlecraft369 commented 2 years ago

Screenshot 2021-09-13 175816

Battlecraft369 commented 2 years ago

這是如何解決？

xinntao commented 2 years ago

我理解 Cupscale GUI 只能使用ncnn来做，不适合直接使用pytorch。

如果需要直接使用pytorch，还是需要安装环境，然后命令行中使用。具体如何解决，我也不是太清楚~

Battlecraft369 commented 2 years ago

我理解 Cupscale GUI 只能使用ncnn来做，不适合直接使用pytorch。

如果需要直接使用pytorch，还是需要安装环境，然后命令行中使用。具体如何解决，我也不是太清楚~

首先感謝老師解答。明白了，不用Cupscale也可以，現在遇到的問題是，我下載了您共享的源碼，裝好了環境也可以進行upscale，但是它在用cpu，沒有用GPU的CUDA。能用什麽方法解決一下嗎？還是説現在放出來的代碼就是使用CPU的？用CPU好慢啊。。。

xinntao commented 2 years ago

pytorch，它应该自动使用GPU的，你可以看下你的GPU是否能被PyTorch识别到