Restoring Vision in Hazy Weather with Hierarchical Contrastive Learning

Wang, Tao; Tao, Guangpin; Lu, Wanglong; Luo, Wenhan; Zhang, Xiaoqin; Lu, Tiecheng

doi:10.48550/arxiv.2212.11473

Cited by 1 publication

(1 citation statement)

References 0 publications

Supporting

Mentioning

Contrasting

Order By: Relevance

“…Furthermore, the results on the nighttime dataset NHR [19] are reported in Table III. Our model outperforms the recent algorithm HCD [33] by a large margin of 2.04 dB PSNR. Fig.…”

Section: ) Image Dehazingmentioning

confidence: 80%

PSNet: Towards Efficient Image Restoration With Self-Attention

Cui,

Knoll

2023

IEEE Robot. Autom. Lett.

View full text Add to dashboard Cite

Image restoration aims to recover a clean image from various degradations, e.g., haze, snow, and blur, playing an important role in robot vision, autonomous vehicles, and medical imaging. Recently, the use of Transformer has witnessed a significant improvement in multifarious image restoration tasks. However, despite a few remedies to reduce the quadratic complexity of self-attention, these approaches are still impractical for real-world applications, which need high efficiency and speed. To ameliorate this issue, we propose an efficient framework for image restoration based on self-attention. To this end, we combine the strengths of patch-based and strip-based self-attention units to improve efficiency. More specifically, we apply self-attention of different operation scales to features of different resolutions, i.e., we adopt a relatively smaller region for self-attention on high-resolution features while a larger region for low-restoration features. In addition, instead of using global self-attention in each partitioned region, we leverage a strip-based version for low complexity. To further improve efficiency, we insert our design into a U-shaped CNN network to establish our framework, dubbed PSNet. Extensive experiments demonstrate that our network receives state-of-the-art performance on five representative image restoration tasks with low computational complexity and high speed, including single-image defocus deblurring, image dehazing, image motion deblurring, image desnowing, and image denoising.

show abstract

“…Furthermore, the results on the nighttime dataset NHR [19] are reported in Table III. Our model outperforms the recent algorithm HCD [33] by a large margin of 2.04 dB PSNR. Fig.…”

Section: ) Image Dehazingmentioning

confidence: 80%

PSNet: Towards Efficient Image Restoration With Self-Attention

Cui,

Knoll

2023

IEEE Robot. Autom. Lett.

View full text Add to dashboard Cite

show abstract

Restoring Vision in Hazy Weather with Hierarchical Contrastive Learning

Cited by 1 publication

References 0 publications

PSNet: Towards Efficient Image Restoration With Self-Attention

PSNet: Towards Efficient Image Restoration With Self-Attention

Contact Info

Product

Resources

About