一种基于机器学习的晴雨检测与处理方法

作者:张灿炜; 梁凌宇; 张通; 何淇昌; 黄张金; 林子珩; 许皓钧; 陈帅琦
来源:2020-12-18, 中国, ZL202011507579.2.

摘要

本发明公开了一种基于机器学习的晴雨检测与处理方法,包括以下步骤:搜集N张无雨图像,同时搜集H张雨滴模板,利用图像合成的方法合成包含N张有雨图像及N张无雨图像构成的数据集,并根据类别设置标签,划分成训练图像与测试图像。将训练图像输入晴雨分类卷积神经网络,进行晴雨分类,并用测试图像进行训练效果检验;构造支持向量机晴雨图像分类器;构建宽度学习神经网络,输入训练图像到该网络,进行宽度学习训练,并使用测试图像检验训练效果;使用集成学习方法组合所述的晴雨分类卷积神经网络、SVM晴雨图像分类器和宽度学习神经网络,形成并行联合判别测试模型;使用该模型检测图像是否有雨,若无雨直接输出;若有雨则进行去雨处理。