EAA-Net: A novel edge assisted attention network for single image dehazing

摘要

Traditional dehazing convolutional neural networks (CNNs) learn the feature maps only from hazy images to the corresponding hazy-free ones, which usually lead to some important feature loss like texture information. The paper proposes an effective edge assisted attention network (EAA-Net) for single image haze removal, aiming to preserve the texture information and improve the overall quality of the dehazed outcomes. The proposed EAA-Net mainly consists of three parts, i.e., a dehaze branch (DB), an edge branch (EB), and a feature fusion residual block (FFRB). In order to solve the loss of the texture information during the dehazing process, the forward information of the EB is concatenated with the DB’s information, and then the fusion result is passed to the FFRB. In addition, the multilevel information boost module is employed into the dehaze branch as a dehaze unit, and it can enhance the multilevel feature information for a better dehazing effect. Extensive experiments conducted on benchmark datasets demonstrate that the proposed EAA-Net performs favorably against some recent state-of-the-art approaches, which includes a better capacity of detail and color maintaining, and competitive consequences both quantitatively and qualitatively.