本文介绍了一种基于平均回复随机微分方程(Mean-Reverting Stochastic Differential Equations)的通用图像复原方法。该方法通过构建一个将高质量图像转换为低质量图像的前向SDE过程,并通过模拟相应的反向SDE过程来实现图像复原,在多项图像复原任务上取得了优秀的性能。
Diffrax是一个基于JAX的数值微分方程求解库,支持ODE/SDE/CDE等多种微分方程类型,提供多种求解器和自动微分功能,可在GPU上高效运行。
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号