Our new X account is live! Follow @wizwand_team for updates
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
113
Low-light Image EnhancementLOL real v2 (test)
PSNR28.342
104
Low-light Image EnhancementLOL syn v2
PSNR29.38
87
Low-light Image EnhancementLOL real v2
PSNR27.8
83
Low-light Image EnhancementLOL Syn v2 (test)
PSNR26.671
78
Low-light Image EnhancementSDSD
PSNR28.42
30
Low-light Image EnhancementLSRW (Huawei) 1.0 (test)
PSNR21.06
14
Low-light Image EnhancementLSRW Nikon 1.0 (test)
PSNR17.69
13
Low-light Image EnhancementHardware Efficiency Benchmark (test)
Latency (GPU) (ms)7.9
5
Showing 9 of 9 rows

Other info

Code

Follow for update