Journal of Beijing University of Posts and Telecommunications

  • EI核心期刊

Journal of Beijing University of Posts and Telecommunications ›› 2022, Vol. 45 ›› Issue (2): 57-64.doi: 10.13190/j.jbupt.2021-106

• PAPERS • Previous Articles     Next Articles

Incremental Dehazing Algorithm Combining Feature Enhancement and Multi-Scale Loss

WANG Keping1, WEI Jinyang1, YANG Yi1,2, FEI Shumin3, CUI Kefei2   

  1. 1. School of Electrical Engineering and Automation, Henan Polytechnic University, Jiaozuo 454003, China;
    2. Zhengmeiji Hydraulic Electric Control Company Limited, Zhengzhou Coal Mining Machinery Group Company Limited, Zhengzhou 450016, China;
    3. College of Automation, Southeast University, Nanjing 210096, China
  • Received:2021-05-28 Published:2021-12-16

Abstract: To improve the clarity of the haze image and realize the generalization of the dehazing ability, a network structure with feature enhancement and multi-scale loss constraint is proposed, which is trained by an incremental training method. The network consists of a teacher network and a student network. The student network enhances the features by learning the attention information of the labeled samples extracted by the teacher network and uses the multi-scale semantic features of the labeled samples as the soft targets. Besides, a multi-scale semantic feature loss measurement mechanism is established, which cascades with the global pixel difference loss to construct the loss function for feature and pixel levels. According to the incremental training method, the teacher network guides the student network to balance the relationship between the new and old knowledge of different datasets. Thus, the network can improve the generalization ability of the supplementary dataset quickly, while retains the original knowledge. Experiment results show that the proposed algorithm performs well in both subjective visual effects and objective evaluation indicators.

Key words: image dehazing, feature enhancement, multi-scale loss constraint, incremental training

CLC Number: