Frontiers of Optoelectronics, Volume. 17, Issue 3, 28(2024)

Low-light enhancement method with dual branch feature fusion and learnable regularized attention

Sun Yixiang, Ni Mengyao, Zhao Ming, Yang Zhenyu, Peng Yuanlong, and Cao Danhua

Restricted by the lighting conditions, the images captured at night tend to suffer from color aberration, noise, and other unfavorable factors, making it difficult for subsequent vision-based applications. To solve this problem, we propose a two-stage size-controllable low-light enhancement method, named Dual Fusion Enhancement Net (DFEN). The whole algorithm is built on a double U-Net structure, implementing brightness adjustment and detail revision respectively. A dual branch feature fusion module is adopted to enhance its ability of feature extraction and aggregation. We also design a learnable regularized attention module to balance the enhancement effect on different regions. Besides, we introduce a cosine training strategy to smooth the transition of the training target from the brightness adjustment stage to the detail revision stage during the training process. The proposed DFEN is tested on several low-light datasets, and the experimental results demonstrate that the algorithm achieves superior enhancement results with the similar parameters. It is worth noting that the lightest DFEN model reaches 11 FPS for image size of 1224×1024 in an RTX 3090 GPU.

Tools

Get Citation

Copy Citation Text

Sun Yixiang, Ni Mengyao, Zhao Ming, Yang Zhenyu, Peng Yuanlong, Cao Danhua. Low-light enhancement method with dual branch feature fusion and learnable regularized attention[J]. Frontiers of Optoelectronics, 2024, 17(3): 28

Download Citation

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

Category: RESEARCH ARTICLE

Received: Mar. 11, 2024

Accepted: Dec. 11, 2024

Published Online: Dec. 11, 2024

The Author Email:

DOI:10.1007/s12200-024-00129-z

Topics