基于雾浓度分类与暗-亮通道先验的多分支去雾网络
A Multi-Branch Defogging Network Based on Fog Concentration Classification and Dark and Bright Channel Priors
张琪东 1迟静 1陈玉妍 1张彩明2
作者信息
- 1. 山东财经大学计算机科学与技术学院 济南 250014;山东省数字媒体技术重点实验室(山东财经大学) 济南 250014
- 2. 山东省数字媒体技术重点实验室(山东财经大学) 济南 250014;山东大学软件学院 济南 250101
- 折叠
摘要
在图像去雾领域中,目前多数去雾模型难以维持精度与效率的平衡,高精度的模型往往伴随着复杂的网络结构,而简单的网络结构又往往会导致低质量的结果.针对该问题提出一个基于雾浓度分类与暗-亮通道先验的多分支去雾模型,通过对带雾图像分类,使用复杂度不同的网络来处理不同雾浓度的图像,可在保证精度的同时提高计算效率.模型由轻量级雾图像分类器和基于暗-亮通道先验的多分支去雾网络 2部分构成:前者将带雾图像分为轻雾、中雾、浓雾 3类,输出雾浓度标签;后者包含 3个结构相同、宽度不同的分支网络,根据雾浓度标签选择不同的分支网络处理不同雾浓度图像,恢复至无雾图像.提出一个新的雾浓度分类方法以及基于该方法的雾浓度分类损失函数,可根据带雾图像的暗通道特征和恢复难度,结合生成图像质量和模型计算效率,得到对带雾图像合理准确的分类结果,达到去雾效果和算力需求的良好平衡.提出新的暗通道与亮通道先验损失函数,用于约束分支去雾网络,可有效提高去雾精度.实验结果表明,模型能够以更低的网络参数量和复杂度得到更优的去雾结果.
Abstract
In the field of image defogging,it is difficult for most defogging models to maintain a balance between accuracy and efficiency.Specifically,high-precision models are often accompanied by complex network structures,and simple network structures often lead to low-quality results.To address the problem,we propose a multi-branch defogging network based on fog concentration classification and dark and bright channel priors.The model uses the defogging networks with different complexity to handle the images with different fog concentrations,which significantly raises the computational efficiency under ensuring the defogging precision.The model is composed of a lightweight foggy image classifier and a multi-branch defogging network.The classifier divides the foggy images into light,medium and dense foggy images and outputs the fog concentration labels.The multi-branch network contains three branches with the same structure but different widths that process three types of fog images separately.We propose a new fog concentration classification method and a new fog concentration classification loss function.The function combines the dark channel characteristics and defogging difficulty of the foggy image with the defogging precision and computational efficiency of the model,so as to obtain a reasonable fog concentration classification,and consequently achieve a good balance of defogging quality and computing power requirements.We propose a new dark channel prior loss function and a new bright channel prior loss function to constrain the multi-branch defogging network,which effectively enhances the defogging precision.Extensive experiments show that the model is beneficial to get better defogging effect with lower network parameters and complexity.
关键词
图像去雾/雾浓度分类/暗通道先验/亮通道先验/卷积神经网络Key words
image defogging/fog concentration classification/dark channel prior/bright channel prior/CNN引用本文复制引用
基金项目
山东省高等学校青创科技支持计划项目(2020KJN007)
济南市"新高校20条"科研带头人工作室项目(2021GXRC092)
国家自然科学基金重点项目(U1909210)
山东省重点研发计划项目(2019GSF109112)
山东省重点研发计划项目(2021SFGC0102)
出版年
2024
NETL
NSTL
万方数据