基于深度估计辅助的图像去雾方法

作者:王柯俨; 王迪; 陈静怡; 许宁; 王光第; 吴宪云; 李云松
来源:2020-04-24, 中国, ZL202010331563.4.

摘要

本发明公开了一种基于深度估计辅助的图像去雾方法,主要解决现有技术对雾气分布估计差和恢复图像纹理细节丢失的问题。其方案是:在Pytorch框架下分别构建深度估计网络和去雾网络;获取一组无雾图像集J,对J进行深度估计和人工加雾得到深度图像集D和有雾图像集I;利用深度图像集和有雾图像集分别训练深度估计网络和去雾网络,得到训练好的深度估计网络和去雾网络;将需要去雾的图像I-c输入至训练好的深度估计网络,输出估计的深度值D-c;将需要去雾的图像I-c和深度值D-c输入至训练好的去雾网络,输出清晰图像。本发明能很好的恢复图像的细节和色调,且峰值信噪比和结构相似性均高于或接近现有技术,可用于有雾图像的清晰化处理。