基于一般概率分布的驾驶员姿态回归方法

摘要

本发明提供一种基于一般概率分布的驾驶员姿态回归方法，包括：获取驾驶员图像；将所述驾驶员图像输入概率预测网络，得到驾驶员姿态的概率分布函数；所述概率预测网络是基于训练集训练得到的；所述训练集包括一一对应的样本和标签，且所述标签的概率分布与设定的一般概率分布一致；根据所述概率分布函数，计算得到驾驶员姿态预测值和置信度。本发明一方面使得驾驶员姿态回归结果包括预测值和对应概率，从而未后续应用根据概率设置该预测值的权重提供了基础；另一方面能够针对驾驶员姿态的客观不均匀分布，得出更为真实可靠的预测值。