一种基于改进yolov5的飞机起落架检测方法技术

本发明专利技术公开了一种基于改进yolov5的飞机起落架检测方法，方法包括：获取飞机起、落图像，并进行标注，得到飞机起落架数据集；构建改进Yolov5模型，将传统Yolov5模型的检测头解耦，引入焦点损失函数计算置信度损失，同时将原有的位置损失函数进行替换；利用飞机起落架数据集训练改进Yolov5模型，得到训练后的模型；利用训练后的模型，完成飞机起落时，起落架的检测。本发明专利技术有益效果是：整体提升了改进后的YOLOv5模型整体性能，通过制作的飞机起落架数据集进行验证，其准确率从83.6％提升到了97.5％。实现了精准检测飞机和起落架。实现了精准检测飞机和起落架。实现了精准检测飞机和起落架。

【技术实现步骤摘要】
一种基于改进yolov5的飞机起落架检测方法


[0001]本专利技术涉及目标检测领域，尤其涉及一种基于改进yolov5的飞机起落架检测方法。

技术介绍

[0002]计算机视觉领域中，目标检测是其基础任务之一，可以广泛应用于智慧交通、工业检测、人脸识别、军事等领域。
[0003]飞机起落架是飞机起飞和降落的唯一支撑结构，起落架的正常运行可以保障飞机的安全性。国内外的多起飞机事故中，就有一部分是由于起落架无法正常运行引起的。在飞机降落过程中，除了飞机上的指示装备外，还有塔台上的管制员用望远镜对飞机起落架是否正常运行进行人工的监测。若将计算机视觉引入，利用算法自动检测飞机起落架，既可以节约人力资源，也可以减少人为错误导致的飞行事故。因此对飞机起落架进行检测可以及时识别飞机的起落架是否正常打开，对保障飞机的正常起飞和降落具有重要的意义。
[0004]目前利用计算机视觉的方法检测起落架是否放下的研究应用较少，技术相对老旧，误报警率较高。目标检测分为One
‑
stage和Two
‑
stage两大类。
[0005]Two
‑
stage算法第一阶段需要先获取感兴趣区域，第二阶段再识别出感兴趣区域的物体类别和位置信息。
[0006]Yolov5算法是一种one
‑
stage算法，可以直接预测出物体类别和位置信息，但是Yolov5的检测头耦合度高，降低了网络的整体性能。训练过程中存在着正负样本数量不均衡的问题。
[0007]此外，yolov5中的GIOU作为位置坐标loss函数时，在一个候选框包含另一个候选框时，GIOU则会退化成IOU，使得IOU作为损失函数时梯度无法计算的问题再次暴露出来。

技术实现思路

[0008]本专利技术所解决的技术问题为：针对现有YOLOv5技术缺点，如检测头耦合度高，训练过程中正负样本数量不均衡，GIOULoss在一个候选框包含另一个候选框时，会出现退化，导致飞机起落架检测速度慢，检测精度不高的问题，提供一种基于改进yolov5的飞机起落架检测方法，方法包括以下步骤：
[0009]S1、获取飞机起、落图像，并进行标注，得到飞机起落架数据集；
[0010]S2、构建改进Yolov5模型，将传统Yolov5模型的检测头解耦，引入焦点损失函数计算置信度损失，同时将原有的位置损失函数进行替换；
[0011]S3、利用飞机起落架数据集训练改进Yolov5模型，得到训练后的模型；
[0012]S4、利用训练后的模型，完成飞机起落时，起落架的检测。
[0013]本专利技术提供的有益效果是：整体提升了改进后的YOLOv5模型整体性能，通过制作的飞机起落架数据集进行验证，其准确率从83.6％提升到了97.5％。实现了精准检测飞机和起落架。
附图说明
[0014]图1是本专利技术方法流程示意图；
[0015]图2是原YOLOv5效果图；
[0016]图3是改进后的YOLOv5检测效果图。
具体实施方式
[0017]为使本专利技术的目的、技术方案和优点更加清楚，下面将结合附图对本专利技术实施方式作进一步地描述。
[0018]请参考图1，图1是本专利技术方法流程示意图。
[0019]一种基于改进yolov5的飞机起落架检测方法，包括以下步骤：
[0020]S1、获取飞机起、落图像，并进行标注，得到飞机起落架数据集；
[0021]需要说明的是，获取飞机起飞和降落的图像，并对图像中的飞机和飞机的起落架进行标注，最终制作出一个飞机起落架数据集。
[0022]S2、构建改进Yolov5模型，将传统Yolov5模型的检测头解耦，引入焦点损失函数计算置信度损失，同时将原有的位置损失函数进行替换；
[0023]需要说明的是，原本YOLOv5的检测头结构通过一个分支即可完成对飞机和起落架的分类和定位。
[0024]而分类和定位这两个任务所感兴趣的方向不同，分类任务对提取的特征与某一类相近更感兴趣，而定位则对候选框位置和真实框位置更加关注，从而方便进行预测框的修正。将两个任务放在统一分支网络结构进行完成，会使得耦合度高，彼此影响，从而降低准确度。
[0025]因此在本申请中，将Yolov5模型的检测头进行解耦。请参考图2，图2是解耦示意图。图2中左边为原有的Yolov模型的检测头，右边为解耦后的。步骤S2中，Yolov5模型的检测头解耦，具体指：将Yolov5模型的检测头分为两个分支，其中一个分支负责飞机起落架的分类，另一个分支负责飞机起落架的位置定位。解耦后的YOLOv5
‑
Head可以提升模型的整体性能。
[0026]需要说明的是，进行飞机起落架检测时，一张图像就会生成成千上万个候选框，通过设置IOU值，真实框与候选框的IOU值大于设定的阈值则标记为正样本，反之低于该阈值，则标记为负样本。因此，飞机起落架检测会进行密集的采样，而图像中的检测的飞机和起落架数量较少，会造成正负样本严重失衡，以至于降低模型的泛化能力。
[0027]因此将YOLOv5的置信度的损失函数是交叉熵损失函数，将其替换成Focal loss损失函数。通过引入α来控制正负样本的权重和控制难分类样本的权重。以此提升模型的性能。
[0028]焦点损失函数的计算式如式(1)：
[0029][0030]式(1)中S2表示一张特征图的网格数量，B表示一个网格生成的预测框数量，i表示第i个网格，j表示第j个预测框，表示第i个网格的第j个预测框是否有预测目
标，如果有则为1，反之为0。为置信度的真实值，C
i
为置信度的预测值，γ为焦点函数的一个超参数，α为焦点函数的一个权重系数，λ
noobj
为权重系数。
[0031]需要说明的是，本申请中，位置损失函数将原来的GIOULoss替换成CIOULoss。
[0032]GIOULoss主要解决了IOULoss两框不相交时，无法优化的问题。
[0033]但是当一个框包含另一个框时，GIOULoss与IOULoss相同，又暴露出了IOULoss的缺陷。因此将该损失函数替换成CIOULoss，CIOULoss不仅仅考虑到了IOU的面积和两框中心点之间的距离，还考虑到了宽高比的损失，因此CIOULoss更适合用作位置回归损失函数。其计算方式如式2所示。
[0034]CIOULoss＝1
‑
CIOU
ꢀꢀꢀꢀ
(2)
[0035]其中：
[0036][0037][0038][0039]式(2)～式(5)中真实框的高宽为w
t
与h
t
，预测框的高宽为w和h，计算得到的ν可以衡量高宽比，IOU表示真实框和预测框的交并比，β是一个参数，不参与计算梯度值，b_c表示预测框的中心点，t_c表示真实框的中心点，d(b_c,t_c)则表示两个中心本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于改进yolov5的飞机起落架检测方法，其特征在于：包括：S1、获取飞机起、落图像，并进行标注，得到飞机起落架数据集；S2、构建改进Yolov5模型，将传统Yolov5模型的检测头解耦，引入焦点损失函数计算置信度损失，同时将原有的位置损失函数进行替换；S3、利用飞机起落架数据集训练改进Yolov5模型，得到训练后的模型；S4、利用训练后的模型，完成飞机起落时，起落架的检测。2.如权利要求1所述的一种基于改进yolov5的飞机起落架检测方法，其特征在于：步骤S2中，Yolov5模型的检测头解耦，具体指：将Yolov5模型的检测头分为两个分支，其中一个分支负责飞机起落架的分类，另一个分支负责飞机起落架的位置定位。3.如权利要求2所述的一种基于改进yolov5的飞机起落架检测方法，其特征在于：步骤S2中引入焦点损失函数，具体指：将Yolov5中用于计算置信度的交叉熵损失函数，替换为焦点损失函数，焦点损失函数通过引入参数α控制正负样本权重和分类样本权重。4.如权利要求3所述的一种基于改进yolov5的飞机起落架检测方法，其特征在于：焦点损失函数的计算式如式(1)：式(1)中，S2表示一张特征图的网格数量，B表示一个网格生成的预测框数量，i表示第i个网格，j表示第j个预测框，表示第i个网格的第j个预测框是否有预测目标，如果有则为1，反之为0；为置信度的真实值，C
i
为置信度的预测值，γ为焦点函数的一个超参数，α为焦点函数的一个权重系数，λ
noobj
为权重系数。5.如权利要求1所述的一种...

【专利技术属性】
技术研发人员：程家明，向东，徐浩，孔繁东，
申请(专利权)人：武汉兴图新科电子股份有限公司，
类型：发明
国别省市：