Zero-DCE网络的自适应损失函数改进
DOI:
作者:
作者单位:

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

作者简介:

毛经坤 (1976-),男,博士,副教授,硕士生导师,主要从事人工智能、大数据挖掘、机器视觉等方面的研究。

通讯作者:

中图分类号:

基金项目:

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


Improvement on the adaptive loss function for the Zero-DCE network
Author:
Affiliation:

(1.School of Electrical Engineering and Automation, Tianjin University of Technology, Tianjin 300384, China;2.United Power Technology Co., Ltd., Beijing 100039, China;3.School of Artificial Intelligence and Data Science, Hebei University of Technology, Tianjin 300401, China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对轻量化微光增强网络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%.

    参考文献
    相似文献
    引证文献
引用本文

陈林,毛经宇,刘坤,毛经坤. Zero-DCE网络的自适应损失函数改进[J].光电子激光,2024,35(2):135~142

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-08-30
  • 最后修改日期:2022-11-21
  • 录用日期:
  • 在线发布日期: 2024-02-02
  • 出版日期: