arXiv · 2401.15204
LYT-NET: Lightweight YUV Transformer-based Network for Low-light Image Enhancement
Abstract
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
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
A. Brateanu, R. Balmez, A. Avram, C. Orhei, C. Ancuti. 2024-01-26. LYT-NET: Lightweight YUV Transformer-based Network for Low-light Image Enhancement. https://arxiv.org/abs/2401.15204
Cite the original work for its findings. Save a collection to share your selection of sources.