Real-CUGAN 的 Rust 移植

实用技术AIReal-CUGANRust超分
浏览数 - 618发布于 - 2023-12-19 - 02:47

这两天心血来潮,把 Real-CUGAN 移植到 Rust 来了,使用的是 Hugging Face 开源的 Candle 框架,这玩意儿新轮子问题多多啊……

目前支持了 pro 模型的两倍超分,同时支持使用 CPU 推理或使用 CUDA 进行 GPU 推理;目前暂时只支持直接对整张图片进行推理,所以图太大会爆显存,后续计划继续支持把大图切分成小块进行推理来解决这个问题。

仓库地址在这里,欢迎 Star!欢迎 PR!也欢迎关注后续进展!

2 条回复

鲲

5370

#1
发布于 2023-12-20 - 00:05

yur!!!!太厉害了!!快教我 rust!

yurzhang
发布于 2023-12-26 - 13:43

目前已经支持了 pro 模型的两倍与三倍超分,支持了把大图切分成小块推理,支持了带透明背景的图片,支持了一键到任意分辨率!

这个项目已经达到我心里的合格线了,欢迎试用!欢迎反馈!

(。>︿<。) 已经一滴回复都不剩了哦~