3 个月前

ID-Blau:基于隐式扩散重建模糊增强的图像去模糊

ID-Blau:基于隐式扩散重建模糊增强的图像去模糊

摘要

图像去模糊旨在消除动态场景中拍摄图像所引入的不期望模糊。尽管已有大量研究致力于通过优化模型架构来提升去模糊性能,但针对图像去模糊的数据增强方法仍鲜有探索。由于连续运动在图像曝光过程中会引入模糊伪影,我们致力于开发一种突破性的模糊增强方法,通过在连续空间中模拟运动轨迹,生成多样化的模糊图像。本文提出了一种基于隐式扩散的重模糊增强方法(Implicit Diffusion-based reBLurring AUgmentation, ID-Blau),该方法利用一张清晰图像与一个可控制的模糊条件图,生成对应的模糊图像。我们通过像素级的模糊条件图对模糊图像的模糊模式进行参数化,将其方向与强度作为关键特征,以模拟运动轨迹,并在连续空间中隐式表征这些模糊模式。通过采样多样化的模糊条件,ID-Blau能够生成训练数据集中未曾见过的多样化模糊图像。实验结果表明,ID-Blau可生成逼真的模糊图像用于训练,显著提升当前先进去模糊模型的性能。项目源代码已公开,地址为:https://github.com/plusgood-steven/ID-Blau。

代码仓库

plusgood-steven/id-blau
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
deblurring-on-goproID-Blau (Restormer)
PSNR: 33.51
SSIM: 0.965
deblurring-on-goproID-Blau (FFTformer)
PSNR: 34.36
SSIM: 0.970
deblurring-on-goproID-Blau (Stripformer)
PSNR: 33.66
SSIM: 0.966
deblurring-on-hide-trained-on-goproID-Blau (Stripformer)
PSNR (sRGB): 31.50
SSIM (sRGB): 0.944
deblurring-on-hide-trained-on-goproID-Blau (Restormer)
PSNR (sRGB): 31.66
SSIM (sRGB): 0.947
deblurring-on-hide-trained-on-goproID-Blau (FFTformer)
PSNR (sRGB): 31.94
SSIM (sRGB): 0.949
deblurring-on-realblur-j-1ID-Blau (Restormer)
PSNR (sRGB): 33.11
SSIM (sRGB): 0.937
deblurring-on-realblur-j-1ID-Blau (Stripformer)
PSNR (sRGB): 33.77
Params(M): 20
SSIM (sRGB): 0.940
deblurring-on-realblur-j-1ID-Blau (FFTformer)
PSNR (sRGB): 32.88
SSIM (sRGB): 0.934
deblurring-on-realblur-rID-Blau (FFTformer)
PSNR (sRGB): 40.45
SSIM (sRGB): 0.975
deblurring-on-realblur-rID-Blau (Stripformer)
PSNR (sRGB): 41.06
SSIM (sRGB): 0.977
deblurring-on-realblur-rID-Blau (Restormer)
PSNR (sRGB): 40.31
SSIM (sRGB): 0.974

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
ID-Blau:基于隐式扩散重建模糊增强的图像去模糊 | 论文 | HyperAI超神经