Fast Enhancement for Non-Uniform Illumination Images using Light-weight CNNs

重点

暗光增强任务面临着输入图像模糊、信息少、低饱和度、低对比度等退化问题,目前有非常多的方法已经提出来解决问题,但是,并没有一个方法足够轻量化、速度快,以至于将模型部署到移动端上。
全文最最重要的就是说自己的网络轻量化、速度快

作者提出了一种方法:

  • 使用经过预处理(retouching module)的数据集(图选自COCO),以原图+max value of RGB+反 max value of RGB 结合图 作为input,先训练illumination netfushion net,illumination net输出under/over exposed 图,fusion net的输出是权重而不是结果。
  • 固定之前网络参数,将输出的权重和之前的跳接图进行相乘,作为输入restoration net输入,进行训练,得到Neg-Noise /Neg-Artifacts(去除噪声和伪影)并与输入进行跳接,得到结果
  • 其中,总的loss由4种loss组成

在这里插入图片描述

为什么用原图+max value of RGB+ 反 max value of RGB(第一部分)和原图+under/over exposed(第二部分)做输入呢?

这里作者没有细说,但是从自己的理解来看,首先原图和under exposed作为输入自不必说,因为这是最终需要增强的图片,然后over exposed的图像作为输入首先肯定是为了更多的场景考虑,换句话也可以说是不同的light level,或者理解成data augment?同时,暗光增强是需要输出视觉质量高,这肯定包括了不过曝,增加了数据的类型,能够让网络学习到更多信息。

Retouching module和Dataset

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值