3 个月前

基于高效频域的Transformer用于高质量图像去模糊

基于高效频域的Transformer用于高质量图像去模糊

摘要

我们提出了一种高效且有效的图像去模糊方法,该方法通过在频域中探索Transformer的特性,实现高质量的图像去模糊。该方法的动机源于卷积定理:两个信号在空间域中的相关性或卷积运算,等价于其在频域中的逐元素乘积。这一特性启发我们设计了一种基于频域的高效自注意力求解器(Frequency-domain-based Self-Attention Solver, FSAS),通过频域中的逐元素乘积操作来估计缩放点积注意力,而非在空间域中进行复杂的矩阵乘法运算。此外,我们观察到,直接在Transformer中使用传统的前馈网络(Feed-Forward Network, FFN)难以获得理想的去模糊效果。为解决这一问题,我们提出了一种简单而有效的判别性频域前馈网络(Discriminative Frequency-domain-based FFN, DFFN),该网络基于联合图像专家组(Joint Photographic Experts Group, JPEG)压缩算法引入了一种门控机制,能够有区别地判断特征中哪些低频与高频信息应被保留,以促进潜在清晰图像的恢复。我们将所提出的FSAS与DFFN整合为一种非对称网络结构,该结构基于编码器-解码器架构,其中FSAS仅应用于解码模块,以进一步提升去模糊性能。实验结果表明,所提方法在多项指标上均优于当前最先进的去模糊技术。代码将公开于 \url{https://github.com/kkkls/FFTformer}。

代码仓库

kkkls/fftformer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
deblurring-on-hide-trained-on-goproFFTformer
PSNR (sRGB): 31.62
Params (M): 16.6
SSIM (sRGB): 0.9455
deblurring-on-realblur-j-1FFTformer
PSNR (sRGB): 32.62
SSIM (sRGB): 0.9326
deblurring-on-realblur-rFFTformer
PSNR (sRGB): 40.11
SSIM (sRGB): 0.9737
image-deblurring-on-goprofftformer
PSNR: 34.21
Params (M): 16.6
SSIM: 0.969

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于高效频域的Transformer用于高质量图像去模糊 | 论文 | HyperAI超神经