arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.07033cs.CV

SelfPromer:具有深度一致性的自提示去雾Transformer

SelfPromer: Self-Prompt Dehazing Transformers with Depth-Consistency

The Hong Kong Polytechnic University(香港理工大学);Nanjing University of Science and Technology(南京理工大学)

Cong Wang, Jinshan Pan, Wanyu Lin, Jiangxin Dong, Xiao-Ming Wu

AI总结:

本文提出SelfPromer,一种基于深度一致性的自提示Transformer去雾方法,通过深度差异特征生成提示并嵌入注意力机制,在VQGAN编码器-解码器网络中实现迭代自提示推理,在感知指标上优于现有方法。

摘要:

本文提出了一种有效的深度一致性自提示Transformer用于图像去雾。其动机源于一个观察:含有残余雾霾的图像与其清晰对应物的估计深度存在差异。因此,强制去雾图像与清晰图像的深度一致性对于去雾至关重要。为此,我们开发了一种基于雾霾输入图像与对应清晰图像之间深度差异特征的提示,该提示可以引导去雾模型实现更好的恢复。具体来说,我们首先将输入图像提取的深层特征应用于深度差异特征,以生成包含输入中雾霾残余信息的提示。然后,我们提出了一个提示嵌入模块,该模块旨在通过将提示线性添加到深层特征来感知雾霾残余。此外,我们开发了一个有效的提示注意力模块,以更加关注雾霾残余以实现更好的去除。通过将提示、提示嵌入和提示注意力整合到基于VQGAN的编码器-解码器网络中,我们可以实现更好的感知质量。由于推理时清晰图像的深度不可用,且一次前馈执行的去雾图像可能仍包含部分雾霾残余,我们提出了一种新的连续自提示推理方法,该方法可以迭代地纠正去雾模型,以生成更好的无雾图像。大量实验表明,在合成和真实世界数据集上,我们的方法在感知指标(包括NIQE、PI和PIQE)方面优于最先进的方法。

英文摘要:

This work presents an effective depth-consistency self-prompt Transformer for image dehazing. It is motivated by an observation that the estimated depths of an image with haze residuals and its clear counterpart vary. Enforcing the depth consistency of dehazed images with clear ones, therefore, is essential for dehazing. For this purpose, we develop a prompt based on the features of depth differences between the hazy input images and corresponding clear counterparts that can guide dehazing models for better restoration. Specifically, we first apply deep features extracted from the input images to the depth difference features for generating the prompt that contains the haze residual information in the input. Then we propose a prompt embedding module that is designed to perceive the haze residuals, by linearly adding the prompt to the deep features. Further, we develop an effective prompt attention module to pay more attention to haze residuals for better removal. By incorporating the prompt, prompt embedding, and prompt attention into an encoder-decoder network based on VQGAN, we can achieve better perception quality. As the depths of clear images are not available at inference, and the dehazed images with one-time feed-forward execution may still contain a portion of haze residuals, we propose a new continuous self-prompt inference that can iteratively correct the dehazing model towards better haze-free image generation. Extensive experiments show that our method performs favorably against the state-of-the-art approaches on both synthetic and real-world datasets in terms of perception metrics including NIQE, PI, and PIQE.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑