首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于注意力特征融合稠密网络的图像去雾算法
引用本文:孟红记,刘沛谚,胡振伟.基于注意力特征融合稠密网络的图像去雾算法[J].东北大学学报(自然科学版),2022,43(12):1717-1723.
作者姓名:孟红记  刘沛谚  胡振伟
作者单位:(东北大学 信息科学与工程学院, 辽宁 沈阳110819)
基金项目:国家自然科学基金资助项目(51634002).
摘    要:目前主流图像去雾算法输出的结果图像存在颜色失真、边缘模糊的问题.为改善上述问题,提出一种基于深度学习的图像去雾算法,所提算法由两个模块构成:注意力特征融合模块和雾霾模型参数估计模块.注意力特征融合模块用于充分提取雾霾图像的颜色、边缘特征;基于稠密连接空洞卷积自编码器的雾霾模型参数估计模块用于估计雾霾模型的参数,改善网络退化的问题.在浓雾图像、薄雾图像数据集上的实验表明,本文提出的算法有效地实现了图像去雾,与主流的图像去雾算法相比具有更高的结构相似性(SSIM),更低的均方误差(mean-square error,MSE)和边缘误差e○edge.

关 键 词:注意力机制  特征融合  稠密连接  空洞卷积  自编码器  
修稿时间:2021-11-23

Image Dehazing Algorithm Based on Attentional Feature Fusion and Dense Network
MENG Hong-ji,LIU Pei-yan,HU Zhen-wei.Image Dehazing Algorithm Based on Attentional Feature Fusion and Dense Network[J].Journal of Northeastern University(Natural Science),2022,43(12):1717-1723.
Authors:MENG Hong-ji  LIU Pei-yan  HU Zhen-wei
Institution:School of Information Science & Engineering, Northeastern University, Shenyang 110819, China.
Abstract:There are problems of distorted colors and blurred edges in the results of the state-of-the-art image dehazing algorithms. For solving the problems,an image dehazing algorithm based on deep learning is proposed. The proposed algorithm consists of two modules: attentional feature fusion module and haze model parameter estimation module. Attentional feature fusion module is designed to extract the color and edge features of hazy images sufficiently. Haze model parameter estimation module based on densely connected dilated convolution auto encoder is used to estimate the parameter of haze model and deal with the network degeneration in image dehazing. Experiments on images with thin haze and thick haze show that the proposed algorithm performs well on image dehazing, and the proposed dehazing algorithm has higher structural similarity (SSIM), lower mean-square error (MSE), lower edge error e○edge than the state-of-the-art image dehazing algorithms.
Keywords:attentional mechanism  feature fusion  densely connected  dilated convolution  auto encoder  
点击此处可从《东北大学学报(自然科学版)》浏览原始摘要信息
点击此处可从《东北大学学报(自然科学版)》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号