[发明专利]基于多级沙漏结构的单幅图像去雾模型构建及去雾方法

摘要

本发明公开了一种基于多级沙漏结构的单幅图像去雾模型构建及去雾方法，通过设计了去雾模型中的传输图估计网络，利用多个级联的传输图估计模块，将传输图信息分解为全局信息和局部信息，通过多级级联的方式，使得后面的沙漏单元可以更好地融合区域中的细节与整体降质情况之间的关系，对上一级沙漏结构输出的结果进行优化，以准确估计远景的相对距离已经各个景物在原图像中的准确位置，逐步消除因不同尺度特征融合而造成的特征图重影，不连续的现象，自适应地处理不同程度雾霾图像，提高了传输图估计的准确度，从而提高了图像去雾的准确率。

主权项

1.一种基于多级沙漏结构的单幅图像去雾模型构建方法，其特征在于，按照以下步骤执行：步骤1、采集多幅原始图像，获得原始图像集；步骤2、利用散射系数以及大气光参数对所述的原始图像集中的每幅原始图像进行加雾，获得每幅待去雾图像，获得待去雾图像集；获得每幅待去雾图像的参数标签，所述的参数标签包括散射系数以及大气光参数，获得参数标签集；步骤3、将所述的待去雾图像集作为输入，将所述的参数标签集以及原始图像集作为输出，训练网络模型，其中所述的网络模型包括依次串联的估计网络以及融合网络；所述的估计网络包括并联的传输图估计网络以及大气光估计网络；所述的融合网络包括大气光散射模型；所述的传输图估计网络包括多个串联的传输图估计模块，每个所述的传输图估计模块包括沙漏子模块；所述的沙漏子模块包括多个沙漏结构，所述的沙漏结构包括依次设置的一个下采样单元、两个卷积单元以及一个上采样单元，在所述下采样单元与所述上采样单元之间还连接有残差单元；获得去雾模型。