[发明专利]一种编解码网络的训练方法、装置、设备及存储介质

权利要求书

1.一种编解码网络的训练方法，其特征在于，包括：

在执行每轮训练时，获取样本图像；

将所述样本图像输入编解码网络，利用所述编解码网络对所述样本图像顺序执行压缩编码处理和解码恢复处理，得到所述样本图像对应的解码图像；

根据所述样本图像和所述样本图像对应的解码图像，确定所述编解码网络对应的图像损失值和解码损失值，其中，所述根据所述样本图像和所述样本图像对应的解码图像，确定所述编解码网络对应的解码损失值，包括：将所述样本图像和所述样本图像对应的解码图像输入预设的增强网络；通过所述增强网络确定所述样本图像的失真率和所述样本图像对应的解码图像的失真率；根据所述样本图像的失真率和所述解码图像的失真率确定所述编解码网络对应的解码损失值；

根据所述编解码网络对应的图像损失值和解码损失值，确定所述编解码网络对应的网络损失值；

在经过多轮训练，所述编解码网络对应的网络损失值处于预设的网络收敛范围时，确定所述编解码网络收敛。

2.根据权利要求1所述的方法，其特征在于，

所述获取样本图像，包括：获取至少一个样本图像；

所述将所述样本图像输入编解码网络，利用所述编解码网络对所述样本图像顺序执行压缩编码处理和解码恢复处理，得到所述样本图像对应的解码图像，包括：

生成所述至少一个样本图像对应的样本图像矩阵；所述样本图像矩阵中的每一行向量为一个一维的所述样本图像；

将所述样本图像矩阵输入所述编解码网络，利用所述编解码网络对所述样本图像矩阵顺序执行压缩编码处理和解码恢复处理，得到解码图像矩阵；其中，所述解码图像矩阵中的每一行向量为一个所述样本图像对应的一维的解码图像。

3.根据权利要求1所述的方法，其特征在于，

所述编解码网络，包括：图像编码网络和图像解码网络；所述图像编码网络的输出端连接预设的存储装置；所述图像解码网络的输入端连接所述存储装置；

所述将所述样本图像输入编解码网络，利用所述编解码网络对所述样本图像顺序执行压缩编码处理和解码恢复处理，得到所述样本图像对应的解码图像，包括：

将所述样本图像输入所述图像编码网络，利用所述图像编码网络对所述样本图像执行压缩编码处理，得到所述样本图像对应的压缩数据并且将所述样本图像对应的压缩数据输出到所述存储装置中；

通过所述图像解码网络从所述存储装置中获取所述样本图像对应的压缩数据；利用所述图像解码网络对所述样本图像对应的压缩数据执行解码恢复处理，得到所述样本图像对应的解码图像。

4.根据权利要求3所述的方法，其特征在于，在确定所述编解码网络收敛之后，还包括：

接收图像存储指令；所述图像存储指令用于指示存储第一目标图像；

将所述第一目标图像输入所述编解码网络，并且根据所述图像存储指示选择所述编解码网络中的图像编码网络；

利用所述图像编码网络对所述第一目标图像执行压缩编码处理，得到所述第一目标图像对应的压缩数据并且输出到所述存储装置中，以便存储所述第一目标图像。

5.根据权利要求4所述的方法，其特征在于，

所述图像存储指令用于指示存储至少一个第一目标图像；

所述将所述第一目标图像输入所述编解码网络，包括：

生成所述至少一个第一目标图像对应的第一目标图像矩阵；所述第一目标图像矩阵中每一行向量为一个一维的所述第一目标图像；

将所述第一目标图像矩阵输入所述编解码网络；

所述利用所述图像编码网络对所述第一目标图像执行压缩编码处理，得到所述第一目标图像对应的压缩数据，包括：

利用所述图像编码网络对所述第一目标图像矩阵执行压缩编码处理，得到所述第一目标图像矩阵对应的压缩数据矩阵；所述压缩数据矩阵中的每一行向量为一个所述第一目标图像对应的压缩数据。