[发明专利]图像压缩方法、装置、设备及存储介质

权利要求书

1.一种图像压缩方法，其特征在于，包括：

获取原始图像中像素点的第一梯度信息；

根据所述第一梯度信息确定所述原始图像对应的掩码图；

将所述掩码图和所述原始图像输入图像压缩模型中，输出目标图像；其中，所述图像压缩模型基于至少一种损失信息训练获得。

2.根据权利要求1所述的方法，其特征在于，根据所述第一梯度信息确定所述原始图像对应的掩码图，包括：

将所述第一梯度信息与设定阈值进行比较；

若所述第一梯度信息大于所述设定阈值，则将所述第一梯度信息作为所述像素点在掩码图中的像素值；若所述第一梯度信息小于或等于所述设定阈值，则将设定值作为所述像素点在所述掩码图中的像素值。

3.根据权利要求1所述的方法，其特征在于，所述损失信息包括如下至少一种：平滑损失信息、高频损失信息、对抗损失信息、图像质量损失信息、重建损失信息。

4.根据权利要求3所述的方法，其特征在于，所述图像压缩模型的训练方式为：

获取原始样本图的第二梯度信息；

根据所述第二梯度信息确定所述原始样本图对应的掩码样本图；

将所述掩码样本图和所述原始样本图输入所述图像压缩模型，输出目标样本图；

基于所述原始样本图、所述第二梯度信息、所述掩码样本图及所述目标样本图中的至少一项确定至少一个损失信息；

基于所述至少一个损失信息的加权求和结果对所述图像压缩模型进行训练。

5.根据权利要求4所述的方法，其特征在于，基于所述原始样本图、所述第二梯度信息、所述掩码样本图及所述目标样本图中的至少一项确定平滑损失信息，包括：

确定所述目标样本图中像素点的第三梯度信息；

将所述第三梯度信息与所述掩码样本图中对应的掩码信息进行融合，获得像素点的高频掩码信息；

基于所述高频掩码信息确定平滑损失信息。

6.根据权利要求5所述的方法，其特征在于，基于所述原始样本图、所述第二梯度信息、所述掩码样本图及所述目标样本图中的至少一项确定高频损失信息，包括：

确定所述第三梯度信息和所述第二梯度信息的偏差信息；

将所述偏差信息与所述掩码样本图中对应的掩码信息进行融合，获得像素点的高频偏差信息；

基于所述高频偏差掩码信息确定高频损失信息。

7.根据权利要求4所述的方法，其特征在于，基于所述原始样本图、所述第二梯度信息、所述掩码样本图及所述目标样本图中的至少一项确定对抗损失信息，包括：

将所述原始样本图和所述目标样本图分别输入判别器，输出第一判别结果和第二判别结果；

根据所述第一判别结果和所述第二判别结果更新所述判别器；

将所述目标样本图输入更新后的判别器中，输出第三判别结果；

根据所述第三判别结果确定对抗损失信息。

8.根据权利要求4所述的方法，其特征在于，基于所述原始样本图、所述第二梯度信息、所述掩码样本图及所述目标样本图中的至少一项确定图像质量损失信息，包括：

将所述目标样本图输入设定图像质量评估模型，输出图像质量评估信息；

根据所述图像质量评估信息确定图像质量损失信息。

9.根据权利要求4所述的方法，其特征在于，基于所述原始样本图、所述第二梯度信息、所述掩码样本图及所述目标样本图中的至少一项确定重建损失信息，包括：

确定所述原始样本图和所述目标样本图之间的采样损失信息、特征损失信息及偏差损失信息；

将所述采样损失信息、特征损失信息及偏差损失信息进行线性叠加，获得重建损失信息。