高速低耗图像增强
利用小波变换大幅加速扩散模型推理,并设计高频恢复模块提升细节保真度,实现稳定且高效的弱光图像增强。
Hai, Jiang · Luo, Ao · 范浩强 · 韩松臣 · Liu, Shuaicheng
ACM Transactions on Graphics 2023
在小波域中执行扩散过程,利用小波变换的信息保持特性压缩空间维度,显著减少推理步数和计算量,同时不牺牲信息。
训练期间同时进行前向扩散和去噪,使模型学会稳定去噪,避免推理时产生随机混乱的内容,输出结果更加稳定一致。
引入高频恢复模块,利用垂直和水平方向细节补偿对角线方向信息,实现更精细的细节重建,避免细节丢失。