[发明专利]一种图像还原方法、装置、设备和存储介质

说明书

本发明公开了一种图像还原方法、装置、设备和存储介质，包括：提取包含卷曲目标物的图像的特征信息，并对所述特征信息进行特征均衡化处理，得到均衡化特征信息；对所述均衡化特征进行还原处理得到还原特征，连接所述均衡化特征和所述还原特征得到目标还原图像。上述技术方案，通过均衡化处理特征信息得到的均衡化特征，更加注重对包含卷曲目标物的图像的局部特征进行提取，可以获得关于卷曲目标更多的局部特征信息，连接均衡化特征和还原特征得到的目标还原图像，可以包含更多的卷曲目标物的细节信息，进一步根据均衡化特征和还原特征得到的目标还原图像更加接近包含卷曲目标物的图像，进一步提升了卷曲目标物的还原度和还原效果。

技术领域

本发明实施例涉及图像处理技术，尤其涉及一种图像还原方法、装置、设备和存储介质。

背景技术

在图像采集过程中，往往会因为各种不同的因素，导致采集到的图像中目标物存在遮挡或者偏差，此时需要基于图像还原技术对存在遮挡或者偏差的图像进行修复。

对于包含卷曲物体的图像来说，如果需要还原其图像轮廓和图像纹理。现有的图像还原方法，可以基于边缘纹理和内容纹理对扣除掉一部分的图像进行边缘修复和内容修复，但是无法对包含卷曲物体的图像进行展开，还原度不够，而且还原效果也较差。

所以，亟需一种图像还原方法，实现对包含卷曲物体的还原，并且提升还原效果。

发明内容

本发明提供一种图像还原方法、装置、设备和存储介质，以实现对包含卷曲目标物的图像的还原，并且提升还原效果。

第一方面，本发明实施例提供了一种图像还原方法，包括：

提取包含卷曲目标物的图像的特征信息，并对所述特征信息进行特征均衡化处理，得到均衡化特征；

对所述均衡化特征进行还原处理得到还原特征，连接所述均衡化特征和所述还原特征得到目标还原图像。

进一步地，提取包含卷曲目标物的图像的特征信息，并对所述特征信息进行特征均衡化处理，得到均衡化特征，包括：

基于预设生成对抗模型的生成器提取包含卷曲目标物的图像的特征信息，并对所述特征信息进行特征均衡化处理，得到均衡化特征。

进一步地，所述预设生成对抗模型的生成器包括编码模块；

相应地，基于预设生成对抗模型的生成器提取包含卷曲目标物的图像的特征信息，并对所述特征信息进行特征均衡化处理，得到均衡化特征，包括：

通过所述编码模块提取所述包含卷曲目标物的图像的至少一层特征信息；

根据注意力机制在通道维度和空间维度上对所述特征信息进行特征均衡化，得到均衡化特征。

进一步地，通过所述编码模块提取所述包含卷曲目标物的图像的至少一层特征信息，包括：

对所述编码模块的深层卷积层提取到的特征信息进行特征融合，得到深层特征；

对所述编码模块的浅层卷积层提取到的特征信息进行特征融合，得到浅层特征。

进一步地，在通过所述编码模块提取所述包含卷曲目标物的图像的至少一层特征信息之后，还包括：

拼接所述深层特征和所述浅层特征，得到拼接特征；

对所述拼接特征进行卷积处理，得到融合特征；

其中，根据注意力机制在通道维度和空间维度上对所述融合特征进行特征均衡化，得到均衡化特征。

进一步地，所述预设生成对抗模型通过如下方式确定：

获取包含卷曲目标物的训练图像和所述训练图像对应的真实展开图像；