+高级检索
基于双支特征联合映射的端到端图像去雾算法
DOI:
作者:
作者单位:

作者简介:

通讯作者:

基金项目:


End-to-end Image Dehazing Algorithm Based on Joint Mapping of Two-Branch Features
Author:
Affiliation:

Fund Project:

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

    针对卷积神经网络去雾算法中模型复杂度高、特征提取性能差等问题,本文提出了一种基于双支特征联合映射的端到端图像去雾算法.首先对大气散射模型进行变形转换,分离出模型中的双支特征;然后根据双支特点设计了两个特征提取网络MPFEM和SPFEM,分别使用两种注意力机制对其输出特征进行加权;最后将提取到的双支特征输入复原模块恢复清晰图像,并对其进行色彩增强得到最终复原效果.在模型训练过程中为避免使用单一损失函数导致纹理细节丢失等问题,采用多尺度结构相似度和平均绝对误差加权作为损失函数.实验表明,本文所提算法网络结构简单,去雾效果明显,复原图像色彩亮度保真,边缘保持性强.

    Abstract:

    To address the issues of high model complexity and poor feature extraction performance in Convolutional neural network-based dehazing algorithms, this paper proposes an end-to-end image dehazing algorithm based on joint mapping of two-branch features. Firstly, the atmospheric scattering model is transformed to separate the mixed-parameter feature and the single-parameter feature model. Then two feature extraction networks, MPFEM and SPFEM are designed according to the two-branch features and the outputs are weighted by two attention mechanisms. Finally, the extracted two-branch features are sent to the restoration module to restore the clear image and perform color-enhancing to obtain the final restored effect. To avoid the loss of texture details caused by using a single loss function in the model training process, multi-scale structure similarity and mean absolute error weighting are used as the loss function. Experimental results show that the proposed algorithm has a simple network structure, obvious dehazing effect, accurate color brightness restoration, and strong edge preservation.

    参考文献
    相似文献
    引证文献
文章指标
  • PDF下载次数:
  • HTML阅读次数:
  • 摘要点击次数:
  • 引用次数:
引用本文

杨燕 ,陈阳 ?.基于双支特征联合映射的端到端图像去雾算法[J].湖南大学学报:自然科学版,2024,(6):10~19

复制
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-07-05
  • 出版日期:
作者稿件一经被我刊录用,如无特别声明,即视作同意授予我刊论文整体的全部复制传播的权利,包括但不限于复制权、发行权、信息网络传播权、广播权、表演权、翻译权、汇编权、改编权等著作使用权转让给我刊,我刊有权根据工作需要,允许合作的数据库、新媒体平台及其他数字平台进行数字传播和国际传播等。特此声明。
关闭