[发明专利]一种基于卷积神经网络的图像去雾方法

摘要

本发明公开了一种基于卷积神经网络的图像去雾方法，包括以下步骤：S1、将待去雾的有雾图输入到预训练好的透射率估计模型中，得到透射率图；S2、根据所得透射率图和预设的透射率阈值，得到有雾图的薄雾区域掩码图和浓雾区域掩码图；S3、根据薄雾区域掩码图和浓雾区域掩码图将待去雾的有雾图分割为薄雾图和浓雾图；S4、将所得薄雾图和浓雾图同时输入到预训练好的去雾模型进行去雾，得到无雾图。通过估计雾图的透射率图，将有雾图像分为薄雾区域和浓雾区域，同时分别输入到去雾模型中的去薄雾网络和去浓雾网络中，对不同浓度的区域采用不同的参数进行处理，所得结果在去雾模型中进行像素级的合并，得到完整清晰的去雾图，去雾效果较好。