1.基于YOLO的关键点目标检测方法,其特征在于包括以下步骤:步骤一、数据集的制作与处理:
在原始标注框为水平矩形框的标注数据集上,添加各个关键点到标注框左上角顶点的偏移距离(Δx,Δy),标注框左上角的顶点位置坐标为(LUx,LUy),LUx小于所有关键点的x方向上的值,LUy小于所有关键点的y方向上的值,此时,各个关键点位置均为:以标注框左上顶点为坐标轴原点时坐标轴的第四象限;
步骤二、基于预测框左上角顶点偏移量的点目标检测:首先通过YOLO得到预测框,同时得到各个关键点与预测框左上角顶点的偏移量,将网络输出的对应各个关键点的偏移量(Δx,Δy)与预测框左上角顶点的坐标(LUx,LUy)相加,即可得到关键点的坐标位置;
所述步骤一中,关键点的个数为4,各个关键点相对于预测框左上角的偏移量的距离的公式如(1.8)~(2.1)所示,模型对每个预测框,会输出tD1
D2
D3
D4
其中:D1
YOLO点目标检测中关键点的损失函数如公式(2.2)所示,该式子中关键点个数为4:若关键点增多,关键点损失函数将如(2.3)所示,式子中m为关键点的数量:YOLO点目标检测是在原始YOLOv3的检测中增加了关键点的计算损失,因此最终损失函数为:Loss
2.根据权利要求1所述基于YOLO的关键点目标检测方法,其特征在于:所述步骤二中,图片输入到YOLO网络中,得到预测框的同时得到关键点到预测框左上角的偏移量,根据预测框左上角顶点和4个关键点的偏移量,计算出关键点所在的位置,再将关键点进行连接,得到精准的定位框。