【技术实现步骤摘要】
一种图像处理方法、装置、电子设备及存储介质
本专利技术涉及计算机
,尤其涉及一种图像处理方法、装置、电子设备及存储介质。
技术介绍
雨是一种自然现象,天空中的雨线落到玻璃或者镜头上,会变成雨滴,在阴雨天气拍照时,拍出的图片很容易带有雨水(例如雨线、雨滴),这会严重影响图片的美观。目前的图像去雨方法主要是关注图像中的雨滴或者雨线,将带雨滴的图片输入神经网络中,使用卷积神经网络提取图片信息,使用重建的损失函数作为监督信号,来对图片进行去雨处理。然而,在实际应用中发现,通过目前的图像去雨方法处理得到的图像中仍然会有雨滴或雨线残留,难以彻底地去除图像中的雨线和雨滴,导致图像去雨的效果不佳。
技术实现思路
本专利技术实施例提供一种图像处理方法、装置、电子设备及存储介质,可以同时去除图片中的雨线和雨滴,提升图像的去雨效果。第一方面,本专利技术实施例提供了一种图像处理方法,所述方法包括:获取待处理图片。对所述待处理图片进行特征检测,以获取所述待处理图片的雨线特征图和雨滴特征图,其中,所述雨 ...
【技术保护点】
1.一种图像处理方法,其特征在于,所述方法包括:/n获取待处理图片;/n对所述待处理图片进行特征检测,以获取所述待处理图片的雨线特征图和雨滴特征图,其中,所述雨线特征图包括所述待处理图片中雨线对应的图像区域,所述雨滴特征图包括所述待处理图片中雨滴对应的图像区域;/n利用所述雨线特征图和所述雨滴特征图对所述待处理图片进行去雨处理,以得到去除雨线和雨滴的目标图片。/n
【技术特征摘要】
1.一种图像处理方法,其特征在于,所述方法包括:
获取待处理图片;
对所述待处理图片进行特征检测,以获取所述待处理图片的雨线特征图和雨滴特征图,其中,所述雨线特征图包括所述待处理图片中雨线对应的图像区域,所述雨滴特征图包括所述待处理图片中雨滴对应的图像区域;
利用所述雨线特征图和所述雨滴特征图对所述待处理图片进行去雨处理,以得到去除雨线和雨滴的目标图片。
2.根据权利要求1所述的方法,其特征在于,所述对所述待处理图片进行特征检测,以获取所述待处理图片的雨线特征图和雨滴特征图,包括:
对所述待处理图片进行特征检测,得到所述待处理图片的特征信息;
对所述特征信息进行解析处理,以获取所述待处理图片中的雨线图像区域和雨滴图像区域;
将所述雨线图像区域作为所述待处理图片的雨线特征图,以及将所述雨滴图像区域作为所述待处理图片的雨滴特征图。
3.根据权利要求2所述的方法,其特征在于,所述对所述待处理图片进行特征检测,得到所述待处理图片的特征信息,包括:
将所述待处理图片输入去雨模型的特征提取网络进行特征检测,得到所述特征提取网络的输出结果;
将所述输出结果作为所述待处理图片的特征信息。
4.根据权利要求3所述的方法,其特征在于,所述特征提取网络包括双注意力检测网络。
5.根据权利要求1~4中任一项所述的方法,其特征在于,所述利用所述雨线特征图和所述雨滴特征图对所述待处理图片进行去雨处理,以得到去除雨线和雨滴的目标图片,包括:
利用所述特征信息和所述雨线特征图对所述待处理图片进行去雨处理,得到去除雨线的图片;
利用所述特征信息和所述雨滴特征图对所述待处理图片进行去雨处理,得到去除雨滴的图片;
根据所述待处理图片、所述去除雨线的图片和所述去除雨滴的图片生成去除雨线和雨滴的目标图片。
6.根据权利要求5所述的方法,其特征在于,所述利用所述特征信息和所述雨线特征图对所述待处理图片进行去雨处理,得到去除雨线的图片,包括:
对所述特征信息和所述雨线特征图进行融合处理,得到雨线融合特征图;
将所述雨线融合特征图输入去雨模型的雨线清理网络,以得到去除雨线的图片。
7.根据权利要求5所述的方法,其特征在于,所述利用所述特征信息和所述雨滴特征图对所述待处理图片进行去雨处理,得到去除雨滴的图片,包括:
对所述特征信息和所述雨滴特征图进行融合处理,得到雨滴融合特征图;
将所述雨滴融合特征图输入去雨模型的雨滴清理网络,以得到去除雨滴的图片。
8.根据权利要求5所述的方法,其特征在于,所述根据所述待处理图片、所述去除雨线的图片和所述去除雨滴的图片生成去除雨线和雨滴的目标图片,包括:
将所述待处理图片、所述去除雨线的图片和所述去除雨滴的图片输入去雨模型的图像合成网络,以得到去除雨线和雨滴的目标图片。
9.根据权利要求6所述的方法,其特征在于,所述对所述特征信息和所述雨线特征图进行融合处理,得到雨线融合特征图,包括:
对所述特征信息和所述雨线特征图进行融合处理,得到第一融合特征图;
获取预置常数与所述雨线特征图之间...
【专利技术属性】
技术研发人员:张凯皓,罗文寒,刘威,
申请(专利权)人:腾讯科技深圳有限公司,
类型:发明
国别省市:广东;44
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。