[发明专利]一种基于深度学习的失真图像重建方法及相关装置

说明书

本申请公开了一种基于深度学习的失真图像重建方法及相关装置，方法包括：根据高清图像集构建失真图像重建训练集，失真图像重建训练集包括预置失真图像和对应的预置图像重建矩阵；采用失真图像重建训练集对初始DFISTA网络模型进行网络层梯度式迭代训练，得到预设DFISTA网络模型，初始DFISTA网络模型根据FISTA算法和深度神经网络构建；通过预设DFISTA网络模型对当前失真图像进行图像重建，得到失真图像重建矩阵。模型通过梯度迭代方式训练，不仅可以确保模型的准确可靠性，还能加快模型的收敛速度。因此，本申请能解决现有技术要么过于复杂，要么缺乏准确性，导致实际的失真图像处理效果较差的技术问题。

技术领域

本申请涉及图像处理技术领域，尤其涉及一种基于深度学习的失真图像重建方法及相关装置。

背景技术

网络发展的趋势是从虚拟到与现实结合，所以视频通话越来越成为人们日常工作与生活最常用到的通信手段之一。通过视频通话，可以更真实的进行交流和表达，也增强了沟通的亲切感。但是视频通话由于网络原因或通话设备卡顿就会出现视频失真问题，影响视频通话的体验。而视频是由多个有渐变规律的图像组成的，当图像每秒变化的次数超过25张后，人眼就不能辨别出每张图像，看到的就是视频；所以，视频失真的本质是图像失真。现有失真图像的处理技术要么过于复杂，会造成较大的计算量；要么就无法确保重建的准确率，导致实际的处理效果较差。

发明内容

本申请提供了一种基于深度学习的失真图像重建方法及相关装置，用于解决现有技术要么过于复杂，要么缺乏准确性，导致实际的失真图像处理效果较差的技术问题。

有鉴于此，本申请第一方面提供了一种基于深度学习的失真图像重建方法，包括：

根据高清图像集构建失真图像重建训练集，所述失真图像重建训练集包括预置失真图像和对应的预置图像重建矩阵；

采用所述失真图像重建训练集对初始DFISTA网络模型进行网络层梯度式迭代训练，得到预设DFISTA网络模型，所述初始DFISTA网络模型根据FISTA算法和深度神经网络构建；

通过所述预设DFISTA网络模型对当前失真图像进行图像重建，得到失真图像重建矩阵。

优选地，所述根据高清图像集构建失真图像重建训练集，所述失真图像重建训练集包括预置失真图像和对应的预置图像重建矩阵，包括：

将高清图像集分别进行失真处理和矩阵表达操作，得到预置失真图像和对应的预置图像重建矩阵；

将所述预置失真图像作为训练输入数据，同时将所述预置图像重建矩阵作为训练输出实际标签，得到失真图像重建训练集。

优选地，所述根据高清图像集构建失真图像重建训练集，所述失真图像重建训练集包括预置失真图像和对应的预置图像重建矩阵，之前还包括：

通过预置高清设备采集多种场景下的清晰图像，得到初始清晰图像集；

对所述初始清晰图像集中的图像进行优化预处理，得到高清图像集。

优选地，所述采用所述失真图像重建训练集对初始DFISTA网络模型进行网络层梯度式迭代训练，得到预设DFISTA网络模型，所述初始DFISTA网络模型根据FISTA算法和深度神经网络构建，包括：

根据FISTA算法和深度神经网络构建初始DFISTA网络模型，所述初始DFISTA网络模型包括初始梯度下降步长；

采用所述失真图像重建训练集对所述初始DFISTA网络模型进行网络层梯度式迭代训练，并计算出中间重建结果向量；

若所述中间重建结果向量满足预设重建需求，则停止训练，得到预设DFISTA网络模型。

本申请第二方面提供了一种基于深度学习的失真图像重建装置，包括：