Apr, 2023

EWT: 单幅图像去噪高效小波变换器

TL;DR提出了一种高效的小波变换器 (Efficient Wavelet Transformer,EWT) 方法,使用离散小波变换 (DWT) 和反小波变换 (IWT) 降采样和升采样以实现图像降噪。同时,还提出了一种新颖的双流特征提取块 (DFEB) 来在不同的级别提取图像特征,缩短模型推断时间和 GPU 存储器使用量。实验证明,该方法提高了原始 Transformer 的速度超过 80%,减少了 GPU 存储器使用量超过 60%,并取得了良好的去噪结果。