一种基于PP-YOLO的风机叶片图像识别方法技术

本发明专利技术公开了一种基于PP

【技术实现步骤摘要】
一种基于PP
‑
YOLO的风机叶片图像识别方法


[0001]本专利技术涉及图像识别
，更具体的说是涉及一种基于PP
‑
YOLO的风机叶片图像识别方法。

技术介绍

[0002]目前，已经有较多的叶片图像识别方法，基于全卷积神经网络的风机叶片表面损伤检测，通过全卷积神经网络进行语义分割，然后通过对裂纹像素点提取计算出损伤区域和损伤级别，进而进行风机叶片表面损伤识别。
[0003]但是，并未对图像进行精确定位，粗略估计，图片干扰因素太多；基于Mask R
‑
CNN的风机叶片缺陷分类及定位，以ResNet101为骨干网络，先在Coco数据集上对模型进行预训练，后通过迁移学习对模型进行微调，将模型应用在风机叶片的缺陷分类、缺陷区域矩形框回归和缺陷区域MASK生成等，形成最终的风机叶片缺陷识别模型，采用“两步走”策略，先确定候选区域，然后再对候选区域进行分类并预测目标物体位置，计算上比较耗时；基于YOLO v3的风机叶片状态检测与分析，通过LabelImg工具对图像进行多分类标注，并通过在Voc数据集上的预训练模型进行训练和微调，最终输出叶片图像识别结果，虽然是采用“一步走”策略，但是在整体性能上略逊于YOLO v4。
[0004]因此，如何预测框是否包含风机、风机位置坐标以及风机损伤类别，进而完成风机叶片损伤的图像识别是本领域技术人员亟需解决的问题。

技术实现思路

[0005]有鉴于此，本专利技术提供了一种基于PP
‑
YOLO的风机叶片图像识别方法；选用ResNet为骨干网络，并且综合较好的训练技巧配置PP
‑
YOLO网络，在保证精准性的同时降低模型对硬件和时间需求，来计算预测框是否包含风机、风机位置坐标以及风机损伤类别，进而完成风机叶片损伤的图像识别。
[0006]为了实现上述目的，本专利技术采用如下技术方案：
[0007]一种基于PP
‑
YOLO的风机叶片图像识别方法，包括以下步骤：
[0008]S1、获取风机叶片图像数据并进行预处理；
[0009]S2、构建PP
‑
YOLO网络模型，具体的：
[0010]所述PP
‑
YOLO网络模型包括骨干网络、检测neck和检测head，所述骨干网络通过检测neck与所述检测head连接，所述骨干网络中采用ResNet50
‑
vd模型替代DarkNet
‑
53，并加入可变卷积层，所述检测head由两个卷积层构成，具体为3*3卷积层和1*1卷积层，所述检测head用于输出预测结果，每个预测结果的输出频道是3(K+5)，K为分类类别数量；
[0011]S3、对构建的模型进行训练；
[0012]S4、对训练都的模型进行评估，并识别叶片异常。
[0013]优选的，所述步骤S1具体包括：
[0014]获取风机叶片在静止和转动状态下的图像数据，对于静止状态下的图像数据，进
行间隔取样，对于旋转状态下的图像数据，首先进行切图处理，然后选取多张有效图像数据进行数据增广，将静止状态的图像数据和旋转状态下增广后的图像数据合并为数据集，作为训练样本，随机改变训练样本用以降低模型对属性的依赖性。
[0015]优选的，所述步骤S2还包括：
[0016]所述输出频道3(K+5)表示为：每个预测位置为3个anchor，每个ancheor依次包括K个分类类别，4个预测框和1个目标预测得分。
[0017]优选的，所述步骤S3具体包括：
[0018]S31、提高每轮训练的参数个数，调整学习率；
[0019]S32、在检测neck时，以一定的几率随机地临时丢弃一部分神经元节点，减少隐藏层的神经元个数，提高模型的拟合度；
[0020]S33、引入IoU Aware loss，增加位置准确率信息进行得分计算。
[0021]优选的，所述步骤S4具体包括：
[0022]以召回率和准确率为行纵坐标，得到二维曲线，即PR曲线，将PR曲线下的面积作为衡量尺度，得到AP值，公式为：
[0023][0024]其中，Q
R
为验证集的个数，AP为每个验证集，即PR曲线下的面积。
[0025]经由上述的技术方案可知，与现有技术相比，本专利技术公开提供了一种基于PP
‑
YOLO的风机叶片图像识别方法；选用ResNet为骨干网络，并且综合较好的训练技巧配置PP
‑
YOLO网络，在保证精准性的同时降低模型对硬件和时间需求，来计算预测框是否包含风机、风机位置坐标以及风机损伤类别，进而完成风机叶片损伤的图像识别。
附图说明
[0026]为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本专利技术的实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据提供的附图获得其他的附图。
[0027]图1附图为本专利技术提供的基于PP
‑
YOLO的风机叶片图像识别流程结构示意图。
[0028]图2附图为本专利技术提供的风机叶片预处理示意图。
[0029]图3附图为本专利技术提供的目标检测模型评估效果示意图。
[0030]图4附图为本专利技术提供的PP
‑
YOLO网络模型结构示意图。
具体实施方式
[0031]下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0032]本专利技术实施例公开了一种基于PP
‑
YOLO的风机叶片图像识别方法，包括以下步骤：
[0033]S1、获取风机叶片图像数据并进行预处理；
[0034]S2、构建PP
‑
YOLO网络模型，具体的：
[0035]所述PP
‑
YOLO网络模型包括骨干网络、检测neck和检测head，所述骨干网络通过检测neck与所述检测head连接，所述骨干网络中采用ResNet50
‑
vd模型替代DarkNet
‑
53，并加入可变卷积层，所述检测head由两个卷积层构成，具体为3*3卷积层和1*1卷积层，所述检测head用于输出预测结果，每个预测结果的输出频道是3(K+5)，K为分类类别数量；骨干网络输入是图片，输出是提取的图像特征，检测neck利用骨干网络提取出来的特征数据，通过特征金字塔网络进行多尺度融合，生成具有强语义信息的特征金字塔；
[0036]S3、对构建的模型进行训练；
[0037]S4、对训练都的本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于PP
‑
YOLO的风机叶片图像识别方法，其特征在于，包括以下步骤：S1、获取风机叶片图像数据并进行预处理；S2、构建PP
‑
YOLO网络模型，具体的：所述PP
‑
YOLO网络模型包括骨干网络、检测neck和检测head，所述骨干网络通过检测neck与所述检测head连接，所述骨干网络中采用ResNet50
‑
vd模型替代DarkNet
‑
53，并加入可变卷积层，所述检测head由两个卷积层构成，具体为3*3卷积层和1*1卷积层，所述检测head用于输出预测结果，每个预测结果的输出频道是3(K+5)，K为分类类别数量；S3、对构建的模型进行训练；S4、对训练都的模型进行评估，并识别叶片异常。2.根据权利要求1所述的一种基于PP
‑
YOLO的风机叶片图像识别方法，其特征在于，所述步骤S1具体包括：获取风机叶片在静止和转动状态下的图像数据，对于静止状态下的图像数据，进行间隔取样，对于旋转状态下的图像数据，首先进行切图处理，然后选取多张有效图像数据进行数据增广，将静止状态的图像数据和旋转状...

【专利技术属性】
技术研发人员：邓茂芝，杨志强，
申请(专利权)人：贰佰号科技北京有限公司，
类型：发明
国别省市：