FOANet: A Feedback Operation-Attention Network for Single Image Haze Removal

Chia Lin Liu, Lei Chen, Ling Lo, Pin Jui Huang, Hong Han Shuai, Wen Huang Cheng*, Ching Hsuan Wang, Fan Chou

*此作品的通信作者

研究成果: Article同行評審

摘要

Single image dehazing has become an important vision task for prevailing image degradation caused by detrimental atmosphere transmission conditions. Attention mechanism has been widely utilized in learning based methods to assist the model to discard redundant information and hence boost the performance. However, existing methods are mainly dealing with either channel-wise or pixel-wise attention, which requires much more parameters when the size of feature maps increases. In this paper, we introduce the operation-wise attention in the proposed Feedback Operation-Attention Network (FOANet) to focus on attaining optimal combination of network operations for image dehazing. Specifically, our model consists of two main steps. First, the extracted features of hazy input image are fed to the novel operation-attention block which can adjust the weight of different operations dynamically to produce the optimal processed features. The operation space of an operation-attention block comprises vanilla and dilated convolutions with different kernel sizes along with max pooling and average pooling. Second, we adopt curriculum learning with feedback mechanism to continually refine the features in a recurrent fashion and generate the haze-free image. The experimental results on both synthetic and realistic subsets of RESIDE dataset have demonstrated our method can perform dehazing favorably against other dehazing algorithms.

原文English
文章編號e101
期刊APSIPA Transactions on Signal and Information Processing
12
發行號4
DOIs
出版狀態Published - 2023

指紋

深入研究「FOANet: A Feedback Operation-Attention Network for Single Image Haze Removal」主題。共同形成了獨特的指紋。

引用此