这两天心血来潮,把 Real-CUGAN 移植到 Rust 来了,使用的是 Hugging Face 开源的 Candle 框架,这玩意儿新轮子问题多多啊……
目前支持了 pro 模型的两倍超分,同时支持使用 CPU 推理或使用 CUDA 进行 GPU 推理;目前暂时只支持直接对整张图片进行推理,所以图太大会爆显存,后续计划继续支持把大图切分成小块进行推理来解决这个问题。
仓库地址在这里,欢迎 Star!欢迎 PR!也欢迎关注后续进展!
21
这两天心血来潮,把 Real-CUGAN 移植到 Rust 来了,使用的是 Hugging Face 开源的 Candle 框架,这玩意儿新轮子问题多多啊……
目前支持了 pro 模型的两倍超分,同时支持使用 CPU 推理或使用 CUDA 进行 GPU 推理;目前暂时只支持直接对整张图片进行推理,所以图太大会爆显存,后续计划继续支持把大图切分成小块进行推理来解决这个问题。
仓库地址在这里,欢迎 Star!欢迎 PR!也欢迎关注后续进展!
2 条回复