基于神经结构搜索和雨密度引导的单幅图像去雨方法

作者:傅予力; 霍万良; 蔡磊; 向友君; 夏君君; 张颖
来源:2022-06-14, 中国, CN202210665296.3.

摘要

本发明公开了一种基于神经结构搜索和雨密度引导的单幅图像去雨方法,包括以下步骤:S1.确定雨密度推理网络和单幅图像去雨网络的结构搜索空间,构建雨密度推理网络和单幅图像去雨网络;S2.将有雨图像输入雨密度推理网络,雨密度推理网络自动搜索网络结构;S3.冻结雨密度推理网络的结构参数,将有雨图像输入雨密度推理网络,训练雨密度推理网络的网络权重;S4.将有雨图像输入雨密度推理网络,雨密度推理网络输出雨密度图,将有雨图像和雨密度图输入单幅图像去雨网络,单幅图像去雨网络自动搜索网络结构;S5.冻结单幅图像去雨网络的结构参数,将有雨图像和雨密度图输入单幅图像去雨网络,训练单幅图像去雨网络的网络权重,冻结其网络权重,获得去雨模型。