基于贝叶斯卷积神经网络的图像去雾方法及存储介质

摘要

本发明公开了一种基于贝叶斯卷积神经网络的图像去雾方法及存储介质，所述方法包括：获取RESIDE数据集，采用所述RESIDE数据集中的ITS数据集作为训练集，SOTS数据集作为测试集。将所述训练集输入至贝叶斯卷积神经网络内，对所述贝叶斯卷积神经网络进行训练得到最优模型。将所述测试集输入至所述最优模型，采用所述最优模型对所述测试集进行端对端的去雾处理，并输出去雾图像。本发明能够有效地规避过拟合问题，增强去雾模型的鲁棒性。