TSN-CA: A Two-Stage Network with Channel Attention for Low-Light Image Enhancement

TL;DR

This paper introduces TSN-CA, a two-stage neural network with channel attention, designed to enhance low-light images by improving brightness, reducing noise, and restoring details, outperforming existing methods on multiple datasets.

Contribution

The paper proposes a novel two-stage network with channel attention for low-light image enhancement, combining HSV space brightness adjustment with RGB space detail restoration.

Findings

Achieves superior brightness enhancement and denoising.

Effectively preserves details and reduces halo artifacts.

Outperforms state-of-the-art methods quantitatively and qualitatively.

Abstract

Low-light image enhancement is a challenging low-level computer vision task because after we enhance the brightness of the image, we have to deal with amplified noise, color distortion, detail loss, blurred edges, shadow blocks and halo artifacts. In this paper, we propose a Two-Stage Network with Channel Attention (denoted as TSN-CA) to enhance the brightness of the low-light image and restore the enhanced images from various kinds of degradation. In the first stage, we enhance the brightness of the low-light image in HSV space and use the information of H and S channels to help the recovery of details in V channel. In the second stage, we integrate Channel Attention (CA) mechanism into the skip connection of U-Net in order to restore the brightness-enhanced image from severe kinds of degradation in RGB space. We train and evaluate the performance of our proposed model on the LOL real-world and synthetic datasets. In addition, we test our model on several other commonly used datasets without Ground-Truth. We conduct extensive experiments to demonstrate that our method achieves excellent effect on brightness enhancement as well as denoising, details preservation and halo artifacts elimination. Our method outperforms many other state-of-the-art methods qualitatively and quantitatively.