一种基于决策树的低复杂度迭代CU快速划分方法、电子设备及存储介质

一种基于决策树的低复杂度迭代CU快速划分方法、电子设备及存储介质，属于图像与视频处理技术领域。为对视频标准中编码器的CU划分算法进行优化，本发明包括选取视频编码标准中的标准序列，提取编码单元CU的帧内CU特征、帧间CU特征，构建训练集和测试集；利用训练集训练二分类决策树，训练后在测试集上测试分类准确率并微调决策树参数，然后使用训练好的二分类决策树对训练集进行数据清洗，得到数据清洗的训练集；利用数据清洗的训练集训练三分类决策树，得到输出类别包括划分、不划分和无法判别的三分类决策树；然后对采集的CU进行特征提取，然后进行二分类决策树预测、三分类决策树预测，得到判别结果，进行CU快速划分。