Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2411001(2023)

Soft Histogram of Gradients Loss: A loss Function for Optimization of the Image Fusion Networks

Yuxin Long, Wenjie Lai, Huaiyuan Zhang, Hongbo Zhang, Chengshi Li, and Ziji Liu*
Author Affiliations
  • College of Optoelectronic Science and Engineering, University of Electronic Science and Technology, Chengdu 611731, Sichuan, China
  • show less

    Image fusion methods based on deep learning have achieved excellent image fusion performance and have been widely used in biometric recognition, automatic driving and target tracking. However, it is still challenging to extract important texture details and preserve information of images. Therefore, a loss function for infrared and visible image fusion networks is presented. We employ the histogram of oriented gradient (HOG) to calculate the loss function. HOG feature can reflect the direction and magnitude of local gradient in the image, and using HOG feature as the loss function can improve the ability of the network to extract image details. We combine HOG loss with multi-scale structural similarity loss, and train NestFuse, Res2Fusion and UNFusion infrared and visible image fusion networks with the designed loss function. On the TNO dataset, our model increases the standard deviation (SD) of fused images by 2.1476%, 1.2273% and 1.4444% respectively, and increases the visual information fiedity (VIF) of fused images by 1.6529%, 1.4936% and 1.2902% respectively. On the RoadScene dataset, our model increases the SD of the fused images by 1.0083%, 1.1669% and 0.7214% respectively, and increases the VIF of the fused images by 1.8093%, 1.8063% and 1.0406% respectively. The experimental results show that the proposed loss function can extract more effective information from the source image.

    Tools

    Get Citation

    Copy Citation Text

    Yuxin Long, Wenjie Lai, Huaiyuan Zhang, Hongbo Zhang, Chengshi Li, Ziji Liu. Soft Histogram of Gradients Loss: A loss Function for Optimization of the Image Fusion Networks[J]. Laser & Optoelectronics Progress, 2023, 60(24): 2411001

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Imaging Systems

    Received: Mar. 16, 2023

    Accepted: Apr. 4, 2023

    Published Online: Nov. 27, 2023

    The Author Email: Liu Ziji (zjliu@uestc.edu.cn)

    DOI:10.3788/LOP230882

    Topics