Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LYT-NET: Lightweight YUV Transformer-based Network for Low-light Image Enhancement

About

This letter introduces LYT-Net, a novel lightweight transformer-based model for low-light image enhancement (LLIE). LYT-Net consists of several layers and detachable blocks, including our novel blocks--Channel-Wise Denoiser (CWD) and Multi-Stage Squeeze & Excite Fusion (MSEF)--along with the traditional Transformer block, Multi-Headed Self-Attention (MHSA). In our method we adopt a dual-path approach, treating chrominance channels U and V and luminance channel Y as separate entities to help the model better handle illumination adjustment and corruption restoration. Our comprehensive evaluation on established LLIE datasets demonstrates that, despite its low complexity, our model outperforms recent LLIE methods. The source code and pre-trained models are available at https://github.com/albrateanu/LYT-Net

A. Brateanu, R. Balmez, A. Avram, C. Orhei, C. Ancuti• 2024

Related benchmarks

TaskDatasetResultRank
Low-light Image EnhancementLOL v1
PSNR27.23
135
Low-light Image EnhancementLOL real v2
PSNR27.8
122
Low-light Image EnhancementLOL real v2 (test)
PSNR28.342
122
Low-light Image EnhancementLOL syn v2
PSNR29.38
118
Low-light Image EnhancementLOL real v2
PSNR21.83
81
Low-light Image EnhancementLOL Syn v2 (test)
PSNR26.671
78
Low-light Image EnhancementSDSD
PSNR28.42
45
Low-light Image EnhancementLOL v1
SSIM0.826
34
Low-light Image EnhancementLOL syn v2
PSNR23.78
17
Low-light Image EnhancementLSRW (Huawei) 1.0 (test)
PSNR21.06
14
Showing 10 of 12 rows

Other info

Code

Follow for update