一种使用语义指导双阶段雨滴去除方法

本发明提供一种使用语义指导双阶段雨滴去除方法，包括准备Raindrop Clarity数据集的训练集和测试集，具体步骤为：使用所述训练集中的图像进行STRRNet网络训练，生成训练模型；将所述训练模型保存到本地文件夹，使用所述测试集中的图像测试所述训练模型的效果，如果满意，则将该训练模型保存为满意的训练模型；将所述满意的训练模型保存到本地文件夹，使用所述满意的训练模型测试有雨图像。本发明将语义指导模块嵌入编码器和解码器连接处，通过将语义信息与图像特征相结合实现更加精细化的恢复，在网络的末端使用背景恢复模块，用来对模型雨滴去除的结果进行二次恢复模糊的背景，将预模型对同一场景的不同帧图像做中值融合，这样可以大幅度去除雨滴残留和雨滴伪影，将中值融合的图像作为伪GT隐式地扩展了训练数据，以一种半监督的方式对模型进行微调，可以大幅度提升模型对无标签图像的雨滴去除效果。