[发明专利]一种基于边缘增强的图像去模糊方法

权利要求书

1.一种基于边缘增强的图像去模糊方法，其特征在于，包括以下步骤：

获取运动模糊图像并基于边缘提取算法对运动模糊图像进行处理，得到运动模糊图像边缘；

将运动模糊图像和运动模糊图像边缘输入到预设的去模糊神经网络进行特征提取和特征融合，得到去模糊图像。

2.根据权利要求1所述一种基于边缘增强的图像去模糊方法，其特征在于，还包括：

对去模糊图像进行质量评估，得到评估结果。

3.根据权利要求2所述一种基于边缘增强的图像去模糊方法，其特征在于，所述边缘提取算法具体通过整体嵌套边缘检测模型实现，所述获取运动模糊图像并基于边缘提取算法对运动模糊图像进行处理，得到运动模糊图像边缘这一步骤，其具体包括：

获取运动模糊图像并将运动模糊图像输入到预训练的整体嵌套边缘检测模型；

经过多级尺度分别输出多个特征图，得到输出信息；

将输出信息在每个输出特征图的通道维度上连接起来，并经过一个融合卷积层得到最终的边缘图像；

根据最终的边缘图像，得到运动模糊图像边缘。

4.根据权利要求3所述一种基于边缘增强的图像去模糊方法，其特征在于，所述预设的去模糊神经网络包括图像主体部分和边缘主体部分，所述图像主体部分包括图像下采样模块和图像上采样模块，所述边缘主体部分包括边缘下采样模块和边缘上采样模块。

5.根据权利要求4所述一种基于边缘增强的图像去模糊方法，其特征在于，所述将运动模糊图像和运动模糊图像边缘输入到预设的去模糊神经网络进行特征提取和特征融合，得到去模糊图像这一步骤，其具体包括：

将运动模糊图像和运动模糊图像边缘输入到预设的去模糊神经网络；

将运动模糊图像经过图像主体部分，得到内容特征；

将运动模糊图像边缘经过边缘主体部分，得到边缘特征；

将内容特征和边缘特征进行融合处理，得到去模糊图像。

6.根据权利要求5所述一种基于边缘增强的图像去模糊方法，其特征在于，所述将运动模糊图像经过图像主体部分，得到内容特征这一步骤，其具体包括：

运动模糊图像经过图像主体部分4级图像下采样模块和4级图像上采样模块分别提取特征图，并在每一级图像下采样模块中将图像下采样模块中对应维度的特征图与图像上采样模块中对应维度的特征图相加，得到长和宽均变为输入图像1/16的最终特征图；根据最终特征图得到内容特征。

7.根据权利要求6所述一种基于边缘增强的图像去模糊方法，其特征在于，所述将运动模糊图像边缘经过边缘主体部分，得到边缘特征这一步骤，其具体包括：

运动模糊图像边缘经过边缘主体部分的4级边缘下采样模块和4级边缘上采样模块分别提取特征图，并在每一级边缘下采样模块中将边缘下采样模块中对应维度的特征图与边缘上采样模块中对应维度的特征图相加，得到长和宽均变为输入图像1/16的最终边缘特征图；

根据最终边缘特征图得到边缘特征。

8.根据权利要求7所述一种基于边缘增强的图像去模糊方法，其特征在于，所述将内容特征和边缘特征进行融合处理，得到去模糊图像这一步骤，其具体包括：

在上采样过程中通过将边缘特征经过一个卷积层转化为通道为1的掩膜，得到边缘特征掩膜信息；

基于像素级相乘的方法将内容特征和边缘特征掩膜信息进行融合，得到去模糊图像。