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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Brateanu, A., Balmez, R., Avram, A., Orhei, C., Ancuti, C.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916943573811200
author Brateanu, A.
Balmez, R.
Avram, A.
Orhei, C.
Ancuti, C.
author_facet Brateanu, A.
Balmez, R.
Avram, A.
Orhei, C.
Ancuti, C.
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2401_15204
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LYT-NET: Lightweight YUV Transformer-based Network for Low-light Image Enhancement
Brateanu, A.
Balmez, R.
Avram, A.
Orhei, C.
Ancuti, C.
Computer Vision and Pattern Recognition
Image and Video Processing
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
title LYT-NET: Lightweight YUV Transformer-based Network for Low-light Image Enhancement
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2401.15204