Infrared and Laser Engineering, Volume. 53, Issue 3, 20230538(2024)

Image highlight removal method based on parallel multi-axis self-attention

Pengyue Li1,2、*, Xinying Xu1, Yandong Tang3,4, Zhaoxia Zhang1, Xiaoxia Han1, and Haifeng Yue2
Author Affiliations
  • 1College of Electrical and Power Engineering, Taiyuan University of Technology, Taiyuan 030024, China
  • 2Taiyuan Heavy Machinery (Group) Company, Taiyuan 030027, China
  • 3State Key Laboratory of Robotics, Shenyang Institute of Automation, Chinese Academy of Sciences, Shenyang 110016, China
  • 4Institutes for Robotics and Intelligent Manufacturing, Chinese Academy of Sciences, Shenyang 110016, China
  • show less
    Figures & Tables(8)
    The overall structure of the proposed network
    The visual results of different methods on SD1 dataset
    The visual results of different methods on RD dataset
    The visual results of different methods on SHIQ dataset
    Comparison of visual results by ablation models for image highlight removal
    • Table 1. Data distribution of three public highlight datasets

      View table
      View in Article

      Table 1. Data distribution of three public highlight datasets

      DatasetRDSD1SHIQ
      Training dataset1800120009825
      Testing dataset2252 0001000
    • Table 2. Comparison of quantitative results of different methods on three public datasets

      View table
      View in Article

      Table 2. Comparison of quantitative results of different methods on three public datasets

      DatasetSD1RDSHIQ
      PSNRSSIMPSNRSSIMPSNRSSIM
      DCSR18.480.8918.600.8027.640.92
      SHRBF8.010.319.660.3521.660.75
      NMF18.410.5816.110.5922.820.62
      SCS9.770.3012.960.2813.470.49
      SHRRI12.920.6612.350.6116.340.69
      Ours22.580.9325.690.8934.220.95
    • Table 3. Ablation experiments of the contribution of the threshold fusion structure and parallel multi-axis self-attention mechanism

      View table
      View in Article

      Table 3. Ablation experiments of the contribution of the threshold fusion structure and parallel multi-axis self-attention mechanism

      ModelM0M1M2M3M4M5
      PSNR30.1530.7831.4633.6733.1534.22
      SSIM0.910.920.930.940.940.95
    Tools

    Get Citation

    Copy Citation Text

    Pengyue Li, Xinying Xu, Yandong Tang, Zhaoxia Zhang, Xiaoxia Han, Haifeng Yue. Image highlight removal method based on parallel multi-axis self-attention[J]. Infrared and Laser Engineering, 2024, 53(3): 20230538

    Download Citation

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

    Category:

    Received: Sep. 19, 2023

    Accepted: --

    Published Online: Jun. 21, 2024

    The Author Email: Li Pengyue (lipengyue@tyut.edu.cn)

    DOI:10.3788/IRLA20230538

    Topics