帧内预测模式的预测方法、装置、存储介质及电子设备与流程

本发明图像处理，具体而言，涉及一种帧内预测模式的预测方法、装置、存储介质及电子设备。

背景技术：

1、(rough mode decision，rmd)粗略模式决策，作为帧内预测的一项技术，会对块划分所有的预测单元(prediction unit，pu)进行粗略计算，从33种预测模式中，如图4所示，通过率失真优化(rate distortion optimization，rdo)计算进而挑选4或5种预测模式，如图5所示，然后送给下一级模块进一步精确计算选择。由于avs3的划分模式复杂，预测单元数量众多，按照上述技术方法进行计算划分，会增加编码器的复杂度，帧内预测模式的确定效率较低。

技术实现思路

1、本发明实施例提供了一种帧内预测模式的预测方法、装置、存储介质及电子设备，以至少解决相关技术中确定帧内预测模式效率低的技术问题。

2、根据本发明实施例的一个方面，提供了一种帧内预测模式的预测方法，包括：获取待预测模式的编码单元以及上述编码单元的参考像素；对上述待预测模式的每个编码单元，根据训练好的模式预测模型获取编码单元对应的模式预测结果；其中，上述模式预测结果用于指示编码单元在所有类别的预测模式下的概率，每类预测模式包括至少一种帧内预测模式；根据上述模式预测结果确定编码单元对应的目标帧内预测模式。

3、根据本发明实施例的另一方面，还提供了一种帧内预测模式的预测装置，包括：第一获取单元，用于获取待预测模式的编码单元以及上述编码单元的参考像素；预测单元，用于对上述待预测模式的每个编码单元，根据训练好的模式预测模型获取编码单元对应的模式预测结果；其中，上述模式预测结果用于指示编码单元在所有类别的预测模式下的概率，每类预测模式包括至少一种帧内预测模式；第一确定单元，用于根据上述模式预测结果确定编码单元对应的目标帧内预测模式。

4、根据本发明实施例的又一方面，还提供了一种电子设备，包括存储器和处理器，上述存储器中存储有计算机程序，上述处理器被设置为通过上述计算机程序执行上述的帧内预测模式的预测方法。

5、根据本发明实施例的又一方面，还提供了一种计算机可读的存储介质，该计算机可读的存储介质中存储有计算机程序，其中，该计算机程序被设置为运行时执行上述帧内预测模式的预测方法。

6、在本发明实施例中，采用了获取待预测模式的编码单元以及上述编码单元的参考像素；对上述待预测模式的每个编码单元，根据训练好的模式预测模型获取编码单元对应的模式预测结果；其中，上述模式预测结果用于指示编码单元在所有类别的预测模式下的概率，每类预测模式包括至少一种帧内预测模式；根据上述模式预测结果确定编码单元对应的目标帧内预测模式的方法，在上述方法中，由于采用训练好的模式预测模型获取编码单元对应的模式预测结果，不仅减少了编码过程中的计算量，加快了编码速度，而且提高了确定帧内预测模式效率，进而解决了相关技术中确定帧内预测模式效率低的技术问题。

技术特征：

1.一种帧内预测模式的预测方法，其特征在于，包括：

2.根据权利要求1所述的方法，其特征在于，所述根据所述模式预测结果确定编码单元对应的目标帧内预测模式，包括：

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

4.根据权利要求3所述的方法，其特征在于，在所述模式预测模型的每次训练过程中，通过如公式(1)所示的损失函数获取所述模式预测模型：

5.根据权利要求3所述的方法，其特征在于，所述方法还包括：

6.根据权利要求1所述的方法，其特征在于，所述对所述待预测模式的每个编码单元，根据训练好的模式预测模型获取编码单元对应的模式预测结果之前，还包括：

7.根据权利要求6所述的方法，其特征在于，所述将所有帧内预测模式划分为多个不同的类别，包括：

8.一种帧内预测模式的预测装置，其特征在于，包括：

9.一种电子设备，包括存储器和处理器，其特征在于，所述存储器中存储有计算机程序，所述处理器被设置为通过所述计算机程序执行所述权利要求1至7中任一项所述的方法。

10.一种计算机可读的存储介质，其特征在于，所述计算机可读的存储介质包括存储的程序，其中，所述程序运行时执行所述权利要求1至7任一项中所述的方法。

技术总结
本发明公开了一种帧内预测模式的预测方法、装置、存储介质及电子设备。其中，上述方法包括：获取待预测模式的编码单元以及所述编码单元的参考像素；对所述待预测模式的每个编码单元，根据训练好的模式预测模型获取编码单元对应的模式预测结果；其中，所述模式预测结果用于指示编码单元在所有类别的预测模式下的概率，每类预测模式包括至少一种帧内预测模式；根据所述模式预测结果确定编码单元对应的目标帧内预测模式。本发明解决了相关技术中确定帧内预测模式效率低的技术问题。

技术研发人员：张鹏,向国庆
受保护的技术使用者：杭州未名信科科技有限公司
技术研发日：
技术公布日：2024/1/13