【技术实现步骤摘要】
本申请涉及图像处理，具体涉及一种编码模式的预测方法、装置、电子设备和存储介质。
技术介绍
1、目前，在视频编码的过程中，通常会采用预测编码技术，来消除像素间的相关性，例如，对参考像素与当前像素的差值进行编码，以达到视频压缩的目的。
2、但是，在选择传统的视频编码模式时，需要对所有的预测模式或部分相关的预测模式进行遍历，才能选择其中最优的预测模式作为最后的处理模式，使得预测过程繁琐，极大地增加了计算复杂度，延长了对视频文件的处理时长。
技术实现思路
1、本申请提供一种编码模式的预测方法、装置、电子设备和存储介质。
2、本申请实施例提供一种编码模式的预测方法，方法包括：获取至少两帧待处理图像的信息，所述两帧待处理图像为连续的前后两帧图像；将至少所述两帧待处理图像的信息输入至编码模式预测网络中进行预测，确定目标编码模式；其中，所述编码模式预测网络为基于多尺寸的像素块对卷积神经网络进行训练获得的网络，所述目标编码模式用于对所述待处理图像进行编码和/或解码。
3、本申请实施例提供一种编码模式的预测装置，其包括：获取模块，被配置为获取至少两帧待处理图像的信息，所述两帧待处理图像为连续的前后两帧图像；预测模块，被配置为将至少所述两帧待处理图像的信息输入至编码模式预测网络中进行预测，确定目标编码模式；其中，所述编码模式预测网络为基于多尺寸的像素块对卷积神经网络进行训练获得的网络，所述目标编码模式用于对所述待处理图像进行编码和/或解码。
4、本申请实施
5、本申请实施例提供了一种可读存储介质，该可读存储介质存储有计算机程序，计算机程序被处理器执行时实现本申请实施例中的任意一种编码模式的预测方法。
6、根据本申请实施例的编码模式的预测方法、设备、电子设备和存储介质，通过获取至少两帧待处理图像的信息，能够明确待处理图像的信息，方便后续对连续的至少前后两帧待处理图像进行处理；将至少两帧待处理图像的信息输入至编码模式预测网络中进行预测，确定目标编码模式，因其中的编码模式预测网络为基于多尺寸的像素块对卷积神经网络进行训练获得的网络，能够降低传统算法循环遍历寻找最优预测编码模式的时间复杂度，从而在使用其中的目标编码模式对待处理图像进行编码和/或解码时，可减少对待处理图像的处理时长，经过训练标签和网络结构合理的设计，可以获得相近或更高的压缩比，保证图像质量的同时，还能够提升视频图像的编码效率。
7、关于本申请的以上实施例和其他方面以及其实现方式，在附图说明、具体实施方式和权利要求中提供更多说明。
本文档来自技高网...【技术保护点】
1.一种编码模式的预测方法，其特征在于，所述方法包括：
2.根据权利要求1所述的方法，其特征在于，所述两帧待处理图像包括：第一帧待处理图像和第二帧待处理图像；
3.根据权利要求2所述的方法，其特征在于，所述第一帧待处理图像对应的像素块尺寸越大，所述目标编码模式预测网络对应的网络层数越多。
4.根据权利要求2所述的方法，其特征在于，所述依据获取到的所述第一帧待处理图像的树编码单元CTU信息，确定所述第一帧待处理图像对应的像素块尺寸，包括：
5.根据权利要求2所述的方法，其特征在于，所述第一帧待处理图像的树编码单元CTU信息，包括：编码单元CU和所述CU的数量；
6.根据权利要求5所述的方法，其特征在于，所述分析结果包括：所述CU中的待编码像素块对应的预测编码模式的出现次数；
7.根据权利要求1至6中任一项所述的方法，其特征在于，所述获取至少两帧待处理图像的信息之前，还包括：
8.根据权利要求7所述的方法，其特征在于，所述依据多个样本图像和多个预设像素块尺寸，对所述卷积神经网络进行训练，获得多个与所述
9.根据权利要求8所述的方法，其特征在于，所述将多个所述待测试样本图像集中的待测试样本图像输入至所述卷积神经网络中进行训练，获得多个与所述预设像素块尺寸对应的编码模式预测网络，包括：
10.根据权利要求9所述的方法，其特征在于，所述待验证编码模式预测网络的输出结果包括：输出图像对应的像素点的预测模式概率值和预设编码协议支持的预设编码模式数量；
11.根据权利要求2所述的方法，其特征在于，所述待处理图像的信息包括：所述待处理图像的像素块信息、与所述像素块信息对应的预测模式、所述预测模式的数量和编码单元CU的划分信息中的至少一种。
12.一种编码模式的预测装置，其特征在于，其包括：
13.一种电子设备，其特征在于，包括：
14.一种可读存储介质，其特征在于，所述可读存储介质存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至11中任一项所述的编码模式的预测方法。
...【技术特征摘要】
1.一种编码模式的预测方法，其特征在于，所述方法包括：
2.根据权利要求1所述的方法，其特征在于，所述两帧待处理图像包括：第一帧待处理图像和第二帧待处理图像；
3.根据权利要求2所述的方法，其特征在于，所述第一帧待处理图像对应的像素块尺寸越大，所述目标编码模式预测网络对应的网络层数越多。
4.根据权利要求2所述的方法，其特征在于，所述依据获取到的所述第一帧待处理图像的树编码单元ctu信息，确定所述第一帧待处理图像对应的像素块尺寸，包括：
5.根据权利要求2所述的方法，其特征在于，所述第一帧待处理图像的树编码单元ctu信息，包括：编码单元cu和所述cu的数量；
6.根据权利要求5所述的方法，其特征在于，所述分析结果包括：所述cu中的待编码像素块对应的预测编码模式的出现次数；
7.根据权利要求1至6中任一项所述的方法，其特征在于，所述获取至少两帧待处理图像的信息之前，还包括：
8.根据权利要求7所述的方法，其特征在于，所述依据多个样本图像和多个预设像素块...
【专利技术属性】
技术研发人员：曹洲，徐科，孔德辉，杨维，
申请(专利权)人：深圳市中兴微电子技术有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。