光电子·激光2024,Vol.35Issue(2) :135-142.DOI:10.16136/j.joel.2024.02.0605

Zero-DCE网络的自适应损失函数改进

Improvement on the adaptive loss function for the Zero-DCE net-work

陈林 毛经宇 刘坤 毛经坤
光电子·激光2024,Vol.35Issue(2) :135-142.DOI:10.16136/j.joel.2024.02.0605

Zero-DCE网络的自适应损失函数改进

Improvement on the adaptive loss function for the Zero-DCE net-work

陈林 1毛经宇 2刘坤 3毛经坤1
扫码查看

作者信息

  • 1. 天津理工大学电气工程与自动化学院,天津 300384
  • 2. 国电联合动力技术有限公司,北京 100039
  • 3. 河北工业大学人工智能与数据科学学院,天津 300401
  • 折叠

摘要

针对轻量化微光增强网络Zero-DCE在处理亮度变化范围较大的微光图像时,存在不同区域亮度增强不一致导致的图像不清晰问题,本文提出了一种基于伽马变换的自适应损失函数,在原损失函数的基础上降低了网络对图像曝光差异的敏感性,明显改善了微光增强效果.该方法通过在卷积神经网络(convolutional neural network,CNN)中添加CBAM模块提高网络对微光图像特征的表达能力,使用网络增强图像灰度平均值与增强特征图均值的对数距离作为伽马变换自适应系数,最后计算网络增强图像和伽马变换后的图像之间的灰度参数距离.实验表明,与原网络相比,改进后的方法处理效果提升显著,其中在图像评价指标方面,均方误差提升9.7%,峰值信噪比提升13.8%,结构相似性提升6.7%.

Abstract

For light-weight low level light intensifying network,blurred image issue caused by inconsistent light intensifying degree in different area can occur when Zero-DCE handles the low level light image with a bigger brightness variation range.This paper introduces a self-adaptive loss function based on γtransform,on the basis of the original loss function,decreases the sensitivity of the network on image exposure difference and dramatically improves the low level light intensifying effect.In this method,CBAM module is added into the convolutional neural network(CNN)to increase the expression ability of the network to low level light image feature,in addition,the logarithm distance between the average value of gray level of the network intensifying image and the average value of intensifying feature image is selected as γ transformed self-adaptive factor,and finally,the gray level parameter distance between network intensifying image and γ transformed image is calculated.The experiment shows that the performance of this method is dramatically improved comparing to the original network,in which in aspect of image evaluation index,the error mean square is increased by 9.7%,the peak signal to noise ratio is increased by 13.8%,and the structure similarity is increased by 6.7%.

关键词

图像增强/自适应/伽马变换/曝光损失函数/卷积神经网络(CNN)/注意力机制

Key words

image enhancement/adaptive/Gamma transform/exposure loss function/convolutional neu-ral network(CNN)/attention mechanism

引用本文复制引用

基金项目

国家自然科学基金面上项目(62173124)

出版年

2024
光电子·激光
天津理工大学 中国光学学会

光电子·激光

CSCD北大核心
影响因子:1.437
ISSN:1005-0086
参考文献量15
段落导航相关论文