Image restoration method, device, equipment and storage medium

Abstract

本申请涉及工业智能检测技术领域，公开了一种图像复原方法、装置、设备及存储介质。本申请通过动态梯度赋权感受模块，可根据中心像素与各邻域像素的梯度值动态获取各邻域像素的通道切片，并拼接生成中心像素的上下文特征，该方式可根据局部特征梯度自适应地调整其信息聚合策略，从而实现对清晰区域的细节保持和对模糊区域的精准增强；动态梯度赋权感受模块与具有全局感受野的线性双向加权键值计算方式结合，可实现全局上下文建模，降低计算复杂度，使得在清晰区域被精确保留的细节特征，能够以线性计算的效率跨越整个图像，去引导和修正模糊区域的特征表达，从而提取出对离焦模糊具有强大抑制能力的深度特征图，实现对离焦模糊图像的精准复原。

Images