3 个月前

基于空间邻接关系学习分层动态性的图像增强

基于空间邻接关系学习分层动态性的图像增强

摘要

在各类实际图像增强应用中,图像退化通常具有非均匀性、非同质性且形式多样,这对具有固定参数的深度网络在推理阶段带来了严峻挑战。受动态深度网络的启发——这类网络能够根据输入条件自适应地调整模型结构或参数,本文提出一种基于暗通道先验(Dark Channel Prior, DCP)引导的分层动态机制,用于图像增强任务。该机制能够实现从局部到全局的模型参数与特征自适应调整,同时保持区域内空间邻接性。具体而言,本方法依次引入通道-空间层级、结构层级和区域层级的动态组件。通道-空间层级动态机制用于捕捉通道与空间维度上的表征变化;结构层级动态机制则通过建模几何变换并扩展采样位置,以更精准地描述变化的局部特征结构。此外,本文提出一种新颖的区域层级动态机制,利用暗通道先验生成具有空间连续性的掩码,用于动态特征的调控。该区域层级动态机制充分利用了退化图像与未退化图像之间的统计差异,从而提升对图像结构的建模能力。同时,基于DCP引导的区域生成机制本身具有天然的空间一致性,有助于有效捕捉图像的局部一致性特征。实验结果表明,所提方法在多个图像增强任务中均达到当前最优性能,包括图像去雾、图像去雨以及低光照图像增强,且生成的图像在视觉效果上令人满意。相关代码已开源,地址为:https://github.com/DongLiangSXU/HDM。

基准测试

基准方法指标
image-dehazing-on-sots-indoorHDM
PSNR: 38.56
SSIM: 0.991

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于空间邻接关系学习分层动态性的图像增强 | 论文 | HyperAI超神经