一种模型训练、目标检测的方法以及装置制造方法及图纸

本说明书公开了一种模型训练、目标检测的方法以及装置。可应用于自动驾驶系统。在训练过程中，采用基于椭圆高斯核函数映射出的标注热图来对预测热图进行监督，使得训练出的所述目标检测模型输出的预测热图中目标对象上各点的热度值呈椭圆形分布，距离椭圆中心越近的点热度值越高，距离椭圆中心越远的点热度值越低，现实世界中车辆的形状更近似于矩形而非正方形或圆形，因此，基于上述方法所训练出的目标检测模型所输出的热度图能够贴切地表现出目标对象的中心点在各位置的分布概率。目标对象的中心点在各位置的分布概率。目标对象的中心点在各位置的分布概率。

【技术实现步骤摘要】
一种模型训练、目标检测的方法以及装置


[0001]本说明书涉及机器学习
，尤其涉及一种模型训练、目标检测的方法以及装置。

技术介绍

[0002]目标检测算法的一种任务是采用几何形状较为简单的检测框定位出目标对象在待检测数据中的位置。
[0003]在一些目标检测算法的思想中，可以通过预测目标对象的关键点代替对目标对象的检测框的预测，例如，预测目标对象的中心点的位置，并预测目标对象的检测框的尺寸，即可得到目标对象的预测检测框。
[0004]其中，目标检测模型可以输出预测热图，在预测热图中，位于目标对象上的每个点的热度值表示该点为所述目标对象的中心点的概率，通常情况下，针对所述预测热图中位于目标对象上的每个点，该点的热度值越大，该点为所述目标对象的中心点的概率越大，然后，可以将预测热图中热度值为极大值的点，作为该点所位于的目标对象的关键点。
[0005]在现有技术中，常常基于目标对象的标注中心点，以及圆形高斯核函数投影得到标注热图(如图1所示)，并采用标注热图对目标检测模型输出的预测热图进行监督。此时，在由目标对象上各点本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种模型训练方法，其特征在于，包括：确定待检测的样本图像；通过待训练的目标检测模型中的特征提取子网，确定所述待检测的样本图像对应的图像特征；将所述特征提取子网输出的图像特征作为目标检测模型的第一预测子网的输入，通过所述第一预测子网，输出所述样本图像的预测热图；获取预先为所述样本图像确定的标注热图，以所述样本图像的预测热图和标注热图之间的差异最小为目标，调整所述目标检测模型中的参数；其中，预先为所述样本图像确定标注热图，具体包括：获取为所述样本图像中所包含的目标对象确定出的标注中心点，基于所述标注中心点，以及指定的椭圆高斯核函数，将所述样本图像中所包含的各点映射至待标注的标注热图中，得到所述样本图像的标注热图。2.如权利要求1所述的训练方法，其特征在于，所述方法还包括：根据所述样本图像的预测热图，确定出所述目标对象在所述预测热图中的若干个预测关键点；将所述特征提取子网输出的图像特征输入至所述目标检测模型的第二预测子网，通过所述第二预测子网，针对每个预测关键点，输出该预测关键点在每个指定方向上距离所述目标对象的预测检测框的预测间距，作为该预测关键点对应的各预测间距；针对每个预测关键点，根据该预测关键点以及该预测关键点对应的各预测间距，确定出为所述目标对象所预测的基于该预测关键点的预测检测框；获取预先确定的所述目标对象在所述标注热图中的标注检测框，以各预测检测框与所述标注检测框之间的差异最小为目标，调整所述目标检测模型中的参数；其中，预先确定所述目标对象在所述标注热图中的标注检测框，具体包括：获取为所述样本图像中所包含的目标对象确定出的标注检测框，以及为所述样本图像中所包含的目标对象确定出的标注中心点，基于所述标注中心点，以及指定的椭圆高斯核函数，将所述样本图像中所包含的标注检测框映射至待标注的标注热图中，得到所述目标对象在所述标注热图中的标注检测框。3.如权利要求2所述的方法，其特征在于，根据所述样本图像的预测热图，确定出所述目标对象在所述预测热图中的若干个预测关键点，具体包括：确定所述预测热图中位于所述目标对象上各点的热度值，其中，针对所述预测热图中位于所述目标对象上的每个点，该点的热度值越大，该点为所述目标对象的中心点的概率越大；从位于所述目标对象上各点中选择出热度值大于预先设定的热度阈值的点，作为所述目标对象上的预测关键点。4.如权利要求2所述的方法，其特征在于，以各预测检测框与所述标注检测框之间的差异最小为目标，调整所述目标检测模型中的参数，具体包括：确定由各预测检测框以及所述标注检测框构成的检测框集合；针对所述检测框集合中的每个检测框，确定该检测框的各指定角点，以各预测检测框和所述标注检测框的对应的指定角点之间的差异最小为目标，调整所述目标检测模型中的
参数，其中，所述指定角点的数量至少为两个。5.如权利要求1所述的方法，其特征在于，所述方法还包括：将所述特征提取子网输出的图像特征作为目标检测模型的第三预测子网的输入，通过所述第三预测子网，预测所述目标对象的检测框相对于指定方向的偏转角度，作为目标对象的预测偏转角度；获取所确定出的所述目标对象的标注偏转角度，以所述目标对象的预测偏转角度和标注偏转角度之间的差异最小为目标，调整所述目标检测模型中的参数。6.如权利要求1所述的方法，其特征在于，确定待检测的样本图像之前，所述方法还包括：获取待检测的点云数据；确定待检测的样本图像，具体包括：将所述点云数据投影至所述点云数据所在空间中的指定平面，得到所述指定平面上的投影图，作为待检测的样本图像；通过待训练的目标检测模型中的特征提取子网，确定所述待检测的样本图像对应的图像特征，具体包括：将所述点云数据所在的空间划分为若干个体素，针对每个体素，对该体素中点云数据进行特征提取，得到所提取出的该体素中的点云特征；根据所提取出的各体素的点云特征，通过待训练的目标检测模型中的特征提取子网，确定出所述指定平面上的点云特征图，作为所确定出的所述待检测的样本图像对应的图像特征。7.如权利要求6所述的方法，其特征在于，所述方法还包括：将所述特征提取子网输出的图像特征作为目标检测模型的第四预测子网的输入，通过所述第四预测子网，预测所述目标对象的检测框高度，作为所述目标对象的预测高度；获取所确定出的所述目标对象的标注高度，以所述目标对象的预测高度和标注高度之间的差异最小为目标，调整所述目标检测模型中的参数。8.一种目标检测方法，其特征在于，包括：确定待检测的目标图像；通过如权利要求1～7任一所述的方法训练出的目标检测模型中的特征提取子网，确定所述目标图像对应的图像特征；将所述特征提取子网输出的图像特征作为目标检测模型的第一预测子网的输入，通过所述第一预测子网，输出所述目标图像的预测热图，其中，针...

【专利技术属性】
技术研发人员：王子扩，李亚蓓，冯阳，
申请(专利权)人：北京三快在线科技有限公司，
类型：发明
国别省市：