HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

You Only Need One Color Space: An Efficient Network for Low-light Image Enhancement

Yan Qingsen ; Feng Yixu ; Zhang Cheng ; Wang Pei ; Wu Peng ; Dong Wei ; Sun Jinqiu ; Zhang Yanning

You Only Need One Color Space: An Efficient Network for Low-light Image
  Enhancement

Abstract

Low-Light Image Enhancement (LLIE) task tends to restore the details andvisual information from corrupted low-light images. Most existing methods learnthe mapping function between low/normal-light images by Deep Neural Networks(DNNs) on sRGB and HSV color space. Nevertheless, enhancement involvesamplifying image signals, and applying these color spaces to low-light imageswith a low signal-to-noise ratio can introduce sensitivity and instability intothe enhancement process. Consequently, this results in the presence of colorartifacts and brightness artifacts in the enhanced images. To alleviate thisproblem, we propose a novel trainable color space, namedHorizontal/Vertical-Intensity (HVI). It not only decouples brightness and colorfrom RGB channels to mitigate the instability during enhancement but alsoadapts to low-light images in different illumination ranges due to thetrainable parameters. Further, we design a novel Color and Intensity DecouplingNetwork (CIDNet) with two branches dedicated to processing the decoupled imagebrightness and color in the HVI space. Within CIDNet, we introduce theLightweight Cross-Attention (LCA) module to facilitate interaction betweenimage structure and content information in both branches, while alsosuppressing noise in low-light images. Finally, we conducted 22 quantitativeand qualitative experiments to show that the proposed CIDNet outperforms thestate-of-the-art methods on 11 datasets. The code is available athttps://github.com/Fediory/HVI-CIDNet.

Code Repositories

fediory/hvi-cidnet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-enhancement-on-sice-gradCIDNet
Average PSNR: 13.446
LPIPS: 0.318
SSIM: 0.648
image-enhancement-on-sice-mixCIDNet
Average PSNR: 13.425
LPIPS: 0.362
SSIM: 0.636
low-light-image-deblurring-and-enhancement-onCIDNet
Average PSNR: 26.572
LPIPS: 0.120
SSIM: 0.890
low-light-image-enhancement-on-dicmCIDNet
BRISQUE: 21.47
NIQE: 3.36
low-light-image-enhancement-on-limeCIDNet
BRISQUE: 16.25
NIQE: 3.03
low-light-image-enhancement-on-lolCIDNet
Average PSNR: 28.141
FLOPS (G): 7.57
LPIPS: 0.079
Params (M): 1.88
SSIM: 0.889
SSIM (sRGB): 0.889
low-light-image-enhancement-on-lolCIDNet-Normal
Average PSNR: 23.500
FLOPS (G): 7.57
LPIPS: 0.086
Params (M): 1.88
SSIM: 0.870
SSIM (sRGB): 0.870
low-light-image-enhancement-on-lol-v2CIDNet
Average PSNR: 24.111
LPIPS: 0.108
SSIM: 0.868
low-light-image-enhancement-on-lol-v2-1CIDNet
LPIPS: 0.045
PSNR: 25.705
SSIM: 0.942
low-light-image-enhancement-on-lolv2CIDNet
Average PSNR: 28.134
LPIPS: 0.101
SSIM: 0.892
low-light-image-enhancement-on-lolv2-1CIDNet
Average PSNR: 29.566
LPIPS: 0.040
SSIM: 0.950
low-light-image-enhancement-on-mefCIDNet
BRISQUE: 13.77
NIQE: 3.11
low-light-image-enhancement-on-npeCIDNet
BRISQUE: 18.92
NIQE: 3.33
low-light-image-enhancement-on-sony-totalCIDNet
Average PSNR: 22.904
LPIPS: 0.411
SSIM: 0.676
low-light-image-enhancement-on-vvCIDNet
BRISQUE: 30.63
NIQE: 2.49

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
You Only Need One Color Space: An Efficient Network for Low-light Image Enhancement | Papers | HyperAI