4 个月前

LYT-NET:轻量级基于YUV变换的低光照图像增强网络

LYT-NET:轻量级基于YUV变换的低光照图像增强网络

摘要

这封信介绍了LYT-Net,一种用于低光照图像增强(LLIE)的新型轻量级基于变压器的模型。LYT-Net由多个层和可拆卸模块组成,包括我们新提出的模块——通道级去噪器(Channel-Wise Denoiser, CWD)和多阶段挤压与激励融合(Multi-Stage Squeeze & Excite Fusion, MSEF),以及传统的变压器模块,即多头自注意力机制(Multi-Headed Self-Attention, MHSA)。在我们的方法中,采用了双路径策略,将色度通道U和V及亮度通道Y作为独立实体处理,以帮助模型更好地进行光照调整和噪声恢复。我们在已建立的LLIE数据集上进行了全面评估,结果表明尽管该模型复杂度较低,但其性能优于近期的LLIE方法。源代码和预训练模型可在https://github.com/albrateanu/LYT-Net 获取。

代码仓库

基准测试

基准方法指标
low-light-image-enhancement-on-lolLYT-Net
Average PSNR: 27.23
FLOPS (G): 3.49
LPIPS: 0.071
Params (M): 0.045
SSIM: 0.853
low-light-image-enhancement-on-lolv2LYT-Net
Average PSNR: 27.80
LPIPS: 0.078
SSIM: 0.873
low-light-image-enhancement-on-lolv2-1LYT-Net
Average PSNR: 29.38
LPIPS: 0.037
SSIM: 0.939

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
LYT-NET:轻量级基于YUV变换的低光照图像增强网络 | 论文 | HyperAI超神经