arXiv · 2210.16057
Semi-UFormer: Semi-supervised Uncertainty-aware Transformer for Image Dehazing
Abstract
Image dehazing is fundamental yet not well-solved in computer vision. Most cutting-edge models are trained in synthetic data, leading to the poor performance on real-world hazy scenarios. Besides, they commonly give deterministic dehazed images while neglecting to mine their uncertainty. To bridge the domain gap and enhance the dehazing performance, we propose a novel semi-supervised uncertainty-aware transformer network, called Semi-UFormer. Semi-UFormer can well leverage both the real-world hazy images and their uncertainty guidance information. Specifically, Semi-UFormer builds itself on the knowledge distillation framework. Such teacher-student networks effectively absorb real-world haze information for quality dehazing. Furthermore, an uncertainty estimation block is introduced into the model to estimate the pixel uncertainty representations, which is then used as a guidance signal to help the student network produce haze-free images more accurately. Extensive experiments demonstrate that Semi-UFormer generalizes well from synthetic to real-world images.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Ming Tong, Yongzhen Wang, Peng Cui, Xuefeng Yan, Mingqiang Wei. 2022-10-28. Semi-UFormer: Semi-supervised Uncertainty-aware Transformer for Image Dehazing. https://arxiv.org/abs/2210.16057
Cite the original work for its findings. Save a collection to share your selection of sources.