一种基于金字塔网络的半监督单幅图像去雨方法及装置

摘要

本申请实施例公开了一种基于金字塔网络的半监督单幅图像去雨方法及装置；所述方法包括：输入合成雨量图和真实雨量图到半监督学习框架，对所述合成雨量图进行分析，提取多尺度的雨量残差，获得标记的隐空间向量；使用监督阶段的金字塔结构来转移真实雨量图，计算真实的雨量残差和未标记的隐空间向量；根据真实的雨量残差和未标记的隐空间向量，计算出结合标记的隐空间向量和未标记的隐空间向量的损失函数；通过多尺度的高斯拉普拉斯金字塔图像分解技术，利用全局损失的最优解，获得清晰去雨量图；本申请实施例多尺度雨量图可以在设计好的子网内同时进行特征的提取和融合，获得了不同尺度下的隐空间特征向量，使生成的伪真实图更加接近真实雨量图，更好地提升了网络的泛化性能。