确定网站建设的目的,贵州省建设厅官方网站考证,公司做网站有用吗,卖普洱茶做网站文章目录前言代码及原文链接主要的点如何进行图像恢复前言
关于扩散模型以及条件扩散模型的介绍#xff0c;大家可以前往我的上一篇博客#xff1a;扩散模型diffusion model用于图像恢复任务详细原理 (去雨#xff0c;去雾等皆可)#xff0c;附实现代码。
SR3是利用扩散模…
文章目录前言代码及原文链接主要的点如何进行图像恢复前言
关于扩散模型以及条件扩散模型的介绍大家可以前往我的上一篇博客扩散模型diffusion model用于图像恢复任务详细原理 (去雨去雾等皆可)附实现代码。
SR3是利用扩散模型进行图像超分辨率研究的它在使用低分辨率图像作为条件来进行反向采样时先将低分辨率图像直接上采样到高分辨率图像。因此它通过一些改进可以作为图像恢复这种输入与输出图像大小相等的任务。
代码及原文链接 原文https://arxiv.org/abs/2104.07636 源代码https://github.com/Janspiry/Image-Super-Resolution-via-Iterative-Refinement 我用它来做图像恢复的代码图像去雨、去雾等恢复任务代码简化的SR3扩散模型有注释及实验流程
主要的点
SR3模型将DDPM中的残差块替换为BigGAN的残差块并将跳跃连接重新缩放 12\frac{1}{\sqrt{2}}21。 增加了剩余块的数量以及不同分辨率下的通道乘法器。 还有一个拼接操作就是将低分辨图像直接上采样后的图像与噪声图像进行拼接从而实现条件噪声估计。
如何进行图像恢复
源代码中包括大量的小文件不适合初学者调试和使用。我根据其核心代码进行简化将一些不必要的代码文件和代码块删除并在Rain13K (就是MPRNet、Restormer等方法使用的去雨数据集) 上进行了实验目前取得不错的表现。可以参考下边的对比。 其他的任务应该也是可以直接使用的只需要修改一下配置文件中的数据集路径即可。需要使用SR3扩散模型做图像去雨、去雾等恢复任务的可以前往图像去雨、去雾等恢复任务代码简化的SR3扩散模型有注释及实验流程。附上一组在Rain13K训练集上训练在Rain100H上测试的结果。源代码中的迭代次数为1, 000, 000我这里是只迭代了 180, 000次后进行测试的结果。
