1、第 49卷 第 11期2023年 11月Computer Engineering 计算机工程基于改进 YOLOv7的液压阀块表面微小缺陷检测季娟娟1,王佳1,陈亚杰2,卢道华1,3(1.江苏科技大学 机械工程学院,江苏 镇江 212100;2.中国船舶重工集团公司 上海船舶设备研究所,上海 200031;3.江苏科技大学 海洋装备研究院,江苏 镇江 212003)摘要:针对液压阀块表面缺陷尺寸微小、对比度低、周围干扰信息多导致的漏检率高、识别准确率低等问题,提出一种基于改进 YOLOv7的液压阀块表面微小缺陷检测算法。在多尺度特征融合模块后引入 CA注意力机制来提高对微小缺陷特征信息的关注度。
2、使用改进的 UpC 多支路上采样结构代替多尺度特征融合模块中的最近邻插值上采样 UpSampling 模块,以丰富微小缺陷的特征信息。利用改进的 ELAN-RepConv 结构代替多尺度特征融合模块中的 ELAN_2 结构,使模型在训练过程中可以学习到更多的特征信息。为了进一步提高算法的鲁棒性与收敛速度,使用离线数据增强融合 Mosaic数据增强的数据增广技术与 K-means+锚框聚类算法来提高算法性能。实验结果表明:该算法在液压阀块表面微小缺陷数据集中平均精度达到 97.6%,较原 YOLOv7算法提高 8.4个百分点,检测速度达到 55.2 frame/s;相较于 YOLOv7 系列中检
3、测精度最高的 YOLOv7-E6E 算法,该算法在参数量减少 75.4%的情况下,平均精度值提高 1.8个百分点。所提算法在保证实时性的前提下能够有效提高检测精度。关键词:YOLOv7算法;液压阀块;缺陷检测;深度学习;注意力机制开放科学(资源服务)标志码(OSID):中文引用格式:季娟娟,王佳,陈亚杰,等.基于改进 YOLOv7的液压阀块表面微小缺陷检测 J.计算机工程,2023,49(11):302-310.英文引用格式:JI J J,WANG J,CHEN Y J,et al.Detection of minor defects on the surface of hydraulic v
4、alve block based on improved YOLOv7 J.Computer Engineering,2023,49(11):302-310.Detection of Minor Defects on the Surface of Hydraulic Valve Block Based on Improved YOLOv7JI Juanjuan1,WANG Jia1,CHEN Yajie2,LU Daohua1,3(1.School of Mechanical Engineering,Jiangsu University of Science and Technology,Zh
5、enjiang 212100,Jiangsu,China;2.Shanghai Marine Equipment Research Institute,China Shipbuilding Industry Group Co.,Ltd.,Shanghai 200031,China;3.Marine Equipment and Technology Institute,Jiangsu University of Science and Technology,Zhenjiang 212003,Jiangsu,China)【Abstract】To address issues such as the
6、 extremely small size of hydraulic valve block surface defects,low contrast,and significant surrounding interference information,which lead to a high leakage detection rate and low recognition accuracy,a detection algorithm of minor defects on the surface of a hydraulic valve block based on improved
7、 YOLOv7 algorithm is proposed.First,a CA attention mechanism is introduced after the multi-scale feature fusion module to improve attention to the feature information of minor defects.Then,the improved UpC multi-branch upsampling structure is used to replace the nearest-neighbor interpolation UpSamp
8、ling in the multi-scale feature fusion module to enrich the feature information of minor defects.Finally,an improved ELAN-RepConv structure is used to replace the ELAN_2 structure in the multi-scale feature fusion module,so that the model can learn more feature information during the training proces
9、s.To improve the robustness and convergence speed of the algorithm further,offline data augmentation,fusing Mosaic data augmentation,and the K-means+clustering anchor box algorithm are used to enhance the performance of the algorithm.The experimental results indicate that the Average Precision(AP)va
10、lue of this algorithm on the dataset of minor defects on the surface of the hydraulic valve block is 97.6%,8.4 percentage points higher than the original YOLOv7 algorithm,and the detection speed reaches 55.2 frame/s.Compared with the YOLOv7-E6E algorithm,which has the highest detection accuracy in t
11、he YOLOv7 series,the AP value is improved by 1.8 基金项目:国家重点研发计划(2018YFC0309100);江苏省重点研发计划(BE2022062)。作者简介:季娟娟(1997),女,硕士研究生,主研方向为机器视觉、深度学习;王 佳,副教授、硕士;陈亚杰,高级工程师、硕士;卢道华,教授、博士、博士生导师。收稿日期:2023-01-11 修回日期:2023-02-22 Email:开发研究与工程应用文章编号:1000-3428(2023)11-0302-09 文献标志码:A 中图分类号:TP391第 49卷 第 11期季娟娟,王佳,陈亚杰,等:基
12、于改进 YOLOv7的液压阀块表面微小缺陷检测percentage points when the number of parameters is reduced by 75.4%.The experimental results show that the improved algorithm can improve detection precision on the premise of ensuring real-time.【Key words】YOLOv7 algorithm;hydraulic valve block;defect detection;deep learning;a
13、ttention mechanismDOI:10.19678/j.issn.1000-3428.00667340概述 液压系统由于具有输出力大、传动可靠、可控性高、使用寿命长等优点1,已被广泛应用于汽车2-3、机器人4-5、航空航天6、工程机械7-9等领域。液压阀块是液压系统中的关键零部件,通过控制压力油在管路中的流动来实现机械动作,在其生产过程中,铣、刨等粗加工操作都会造成液压阀块表面出现划痕、凹坑等缺陷,这些缺陷容易使液压阀块出现渗漏的情况,甚至会影响到机械的行驶安全。为了提前筛选出表面有缺陷的液压阀块,提高产品的合格率,研 究 液 压 阀 块 表 面 微 小 缺 陷 检 测 方 法 具
14、有 重 要意义。目前,许多研究人员采用传统的机器视觉技术进行表面缺陷检测。文献 10 设计一种基于形态学的边缘检测方法,通过使用 Otsu 阈值算法对轴承外圈表面缺陷进行分割,可实现完整的缺陷边缘提取。文献 11 使用改进的霍夫变换与新的区域划分法,针对纹理区域与非纹理区域自动分配不同的缺陷检测方法,对铝型材料表面缺陷进行检测。然而,当工件表面缺陷类别多、背景复杂、各类缺陷表征形式不同、尺寸差异大时,缺陷特征很难量化,无法设定固定的规则来捕捉所有的有效特征,此时传统的机器视觉技术识别精度会大幅下降。与人工检测的主观性和机器视觉检测的局限性不同,基于深度学习的缺陷检测具有精度高、性能好、泛化程度
15、高等优点,其可通过神经网络自主地发现需要提取哪些特征更能适应复杂的检测背景。因此,近年来,基于深度学习的表面缺陷检测方法得到了 广 泛 研 究,并 逐 步 应 用 在 产 品 缺 陷 检 测 领 域。YOLO(You Only Look Once)12-14系列算法是典型的单阶段目标检测算法,与 Faster R-CNN15、SSD16等目标检测算法相比,其检测速度更快,应用范围更广。为了适应不同的应用场景,提高不同对象的检测精度,研究人员陆续提出了各种基于 YOLO 系列算法的改进缺陷检测方法。文献 17 提出一种基于特征增强 YOLO 的表面缺陷检测算法,通过改进的特征金字塔网络来增强多尺
16、度检测层的空间位置相关性,结合深度可分离卷积和密集连接来降低 YOLO模型规模。文献 18 针对复杂的太阳能电池图像背景、可变的缺陷形态,提出一种基于改进 YOLOv5的太阳能电池表面缺陷检测方法,将可变形卷积加入CSP 模块中,引入 ECA-Net19注意力机制,再增加一个小缺陷预测头,从而有效完成太阳能电池表面缺陷检测任务。然而,本文所研究的液压阀块表面缺陷尺寸非常小,以往的 YOLOv3、YOLOv4、YOLOv5 算法对小目标检测效果较差,目前只有最新的 YOLOv720算法在小目标检测中有较好的性能表现。除此之外,液压阀块表面的划痕与刀纹形状非常类似,当划痕与边缘的对比度较低时,容易
17、出现误检、漏检的情况。因此,针对液压阀块表面缺陷的不同特征,本文提出一种基于改进 YOLOv7 的液压阀块表面微小缺陷检测算法,主要改进有以下 4 个方面:1)在多尺度特征融合模块后引入 CA21注意力机制,在增加较少参数的前提下增强对微小缺陷的特征 提 取 能 力;2)将 多 尺 度 特 征 融 合 模 块 中 的UpSampling 替换为改进的 UpC 多支路上采样结构,通过 UpSampling 与反卷积的并行上采样,在增大特征尺寸的同时丰富微小缺陷的特征信息;3)采用改进 的 ELAN(Efficient Layer Aggregation Network)-RepConv 结 构
18、代 替 多 尺 度 特 征 融 合 模 块 中 的ELAN_2 结构,使算法在训练时能提取到更多的特征信息且不影响推理速度;4)采用 K-means+22算法对数据集进行聚类,聚类生成的锚框更符合本文所研究的液压阀块表面微小缺陷的尺寸特征,从而加快模型的收敛速度并提高微小缺陷的检测精度。最后通过多个对比实验与消融实验来验证本文算法的有效性。1YOLOv7算法YOLOv7 有 YOLOv7、YOLOv7-X、YOLOv7-W6、YOLOv7-E6、YOLOv7-D6、YOLOv7-E6E 这 6 种 变体。其中,YOLOv7 的参数量最少,检测速度最快,能够满足用户对高实时性检测的需求。因此,本
19、文选取 YOLOv7作为基础算法进行改进。YOLOv7的网络结构如图 1所示。由图 1可知,YOLOv7的网络结构由 Input、Backbone、Head 这 3 个部分组成。将待检测对象输入 Backbone中,Backbone部分在经过4 个 CBS(Conv+BN+SiLU)操 作 后,使 用 连 续 的ELAN 结构与 DownC 结构完成特征提取与尺寸缩放;在 Backbone 部 分 的 最 后 一 个 特 征 层 后 使 用3032023年 11月 15日Computer Engineering 计算机工程SPPCSPC 结构,增大感受野并优化特征的提取;将SPPCSPC 结构
20、得到的特征层拼接成一个新的特征层,并和从 Backbone 部分提取到的 2 个特征层一起传入多尺度特征融合模块;在多尺度地融合语义信息和空间信息后,输出 3个不同尺度的目标特征层;将得到的 3 个特征层分别通过 RepConv 结构进行辅助训练;通过 YOLO Head中的卷积调整通道数以得到初步的预测结果;对初步的预测结果进行置信度过滤、NMS(Non-Maximum Suppression)等后处理操作,得到最终的输出结果。2改进的 YOLOv7算法 YOLOv7相较于 YOLOv4、YOLOv3等其他一阶段目标检测算法,对小目标具有更好的检测效果,但是,其在目标物都很微小的情况下并不能
21、得到理想的检测结果。因此,针对液压阀块表面缺陷尺寸微小、对比度低等特点,本文对 YOLOv7 进行改进,改进的 YOLOv7算法网络结构如图 2所示。图 1YOLOv7网络结构Fig.1YOLOv7 network structure304第 49卷 第 11期季娟娟,王佳,陈亚杰,等:基于改进 YOLOv7的液压阀块表面微小缺陷检测2.1CA注意力机制液压阀块的表面缺陷尺寸小,在整个图像中占有较少的信息量,背景复杂且存在过多干扰信息。为了提高小目标检测的准确性和减少无关信息的干扰,本文采用注意力机制自适应地聚焦与小目标有关的细节信息,降低对其他信息的关注度。不同于 SE(Squeeze-an
22、d-Excitation)23通道注意力机制只考 虑 内 部 通 道 信 息、CBAM(Convolutional Block Attention Module)24只考虑局部空间位置范围内的信息,CA 注意力机制在增加较少计算量的前提下,既考虑到不同通道之间关系的重要性,同时又考虑到空间位置间的长期依赖关系。因此,本文在多尺度特征融合模块后添加 3 个 CA 注意力机制来提高模型的检测精度。CA 注意力机制结构如图 3所示。为了获取图像宽度和高度上的注意力并对精确位置信息进行编码,CA 注意力机制首先将传进的输入特征使用全局平均池化沿水平方向与垂直方向进行特征编码,获得在宽度和高度 2 个方
23、向上的特征图;接着将宽度方向特征图的输出维数调换后与高度方向特征图进行拼接,拼接后使用 11 的卷积将得到的中间特征图通道数压缩为 C/r(r 为缩减因子);随后使用 BN(Batch Normalization)批量归一化与非线性操作对特征图进行特征映射,映射后使用11 卷积沿着空间维度将特征图分解成 2 个独立的张量,其中,宽度方向特征图的输出维数恢复至输入状态;然后利用 Sigmoid 激活函数分别得到 2 个方向上的注意力权重;最后在原始特征图上通过乘法加权计算,得到在宽度和高度方向上带有注意力权重的特征图。2.2改进的 UpC多支路上采样结构YOLOv7 使用 DownC 结构代替原
24、有的 stride 为2的卷积对特征层进行尺寸减半操作。DownC 结构使用 MaxPooling 和 stride 为 2 的 33 卷积对特征层进行同步下采样。MaxPooling考虑到局部区域的最大值信息,卷积操作考虑到局部区域中所有值的信图 3CA注意力机制结构Fig.3CA attention mechanism structure图 2改进的 YOLOv7网络结构Fig.2Improved YOLOv7 network structure3052023年 11月 15日Computer Engineering 计算机工程息,因此,结合两者的特性做同步下采样并对结果进行特征拼接,可以
25、有效避免特征信息丢失。参考 DownC 结构,本文提出改进的 UpC 多支路上采样结构,如图 4所示。2条支路一开始都使用 11卷积进行通道压缩,减 少 计 算 量;然 后 分 别 使 用 UpSampling 与ConvTranspose2d 对特征图进行上采样;最后在通道上进行特征图拼接。其中:UpSampling 通过邻近点的 值 进 行 插 值 上 采 样,没 有 参 数 可 以 学 习;ConvTranspose2d 为卷积的逆过程,在训练过程中会自动学习最优参数进行上采样。因此,通过上述2 种方式的结合可以提取到更多的特征信息。反卷积计算公式如下:sout_size=sstride
26、(sin_size-1)+kkernel-2 ppadding(1)其中:sstride为卷积步长;kkernel为卷积核大小;ppadding为边界填充;sin_size为输入尺寸;sout_size为输出尺寸。反卷积的具体步骤如下:步骤 1 在输入对象的每个元素后根据sstride-1做补 0扩充操作,若sstride为 1就不补 0。步骤 2 在步骤 1 的基础上按照ppadding再对整体补 0。步骤 3 将步骤 2所得的结果作为真正的输入,并使用sstride为 1的卷积对其进行计算,获得反卷积后的输出结果。2.3改进的 ELAN-RepConv结构ELAN 结构共有 2 条分支,通
27、过 2 条分支的多个特 征 融 合,使 网 络 能 够 学 习 到 更 多 的 特 征 信 息。第 1条分支先使用一个 11卷积做通道数调整,接着使用多个 33 卷积做特征提取;第 2 条分支使用 11卷积减少计算量。在 2 条分支的特征提取结束后,从第 1条分支引出多个特征图,与第 2条分支的特征图拼接在一起得到最后的特征图。本文提出的改进 ELAN-RepConv结构是将多尺度特征融合模块中的 ELAN_2 结构中的部分 33 卷积替换成 RepConv结构。RepConv结构可以在训练时通过多分支的特征提取学习到更多特征;在推理时可通过结构重参化将其转换成卷积核为 33 的Conv+BN
28、+SiLU+BN 单路结构,此时,ELAN-RepConv结构与原来的 ELAN_2 结构基本一致。因此,改进的 ELAN-RepConv结构能够在不增加过多计算量的前提下进一步提升模型的特征学习能力。改进的ELAN-RepConv结构如图 5所示。2.4K-means+锚框聚类YOLOv7 使用的先验框尺寸是通过 K-means25算法在MS COCO数据集上聚类而得到的。MS COCO数据集大约包含 41%的小目标(area3232),其余都是中等目标与大目标。而液压阀块的表面缺陷基本都为微小缺陷,面积约占整个图像的 0.05%,没有大目标的存在,原先设定的先验框不符合本文研究对象的尺寸特
29、性。因此,需要采用聚类算法针对本文所研究的数据集重新生成更利于模型收敛的先验框尺寸。聚类算法中最常用的是 K-means 算法和K-means+算法,聚类中心初始值设定的好坏对聚类结果有着一定的影响。K-means算法随机选取数据集中的 K 个点作为初始聚类中心,其聚类中心的随机初始化可能使聚类结果陷入局部最优状态26,从而产生错误的收敛结果。而 K-means+算法的聚类中心是在选择过程中自动调优的,在 K-means 算法的基础上解决了聚类中心的初始化问题,可以有效地避免局部最优问题。因此,本文采用 K-means+算法对数据集进行聚类,聚类结果如图 6所示(彩色效果见 计算机工程 官网
30、HTML版)。图 4改进的 UpC多支路上采样结构Fig.4Improved UpC multi-branch upsampling structure图 5改进的 ELAN-RepConv结构Fig.5Improved ELAN-RepConv structure306第 49卷 第 11期季娟娟,王佳,陈亚杰,等:基于改进 YOLOv7的液压阀块表面微小缺陷检测3实验结果与分析 3.1实验数据本文所研究的液压阀块图像是从工厂实际生产线上所采集到的,共有 325 张,图像大小为 2 4642 056,含有凹坑(pit)缺陷的图片有 226张,含有划痕(scratch)缺陷的图片有 99张。为
31、了避免过拟合并提高网络的鲁棒性,降低各方面外界因素对检测结果的影响,本文结合离线数据增强与 Mosaic 数据增强操作对数据集进行扩充。考虑到工场实际生产环境中可能会出现光照不均、对比度变化等情况,为了贴合真实场景可能出现的问题,随机融合上下翻转、亮度调节、高斯平滑、对比度调整、添加高斯噪声这 5 种数据增强操作将原有的数据集扩充到 1 950张。其中,含有凹坑缺陷的图片有 1 356 张,含有划痕缺陷的图片有 594 张。扩充完毕后使用 LabelImg图像标注工具对数据集进行标注,手动框选出图像中缺陷的具体位置并标注出对应的类别,标注结束后,缺陷的位置与类别信息会自动地保存在相应的 PAS
32、CAL VOC格式的 xml文件中。根据缺陷类别将数据集按照 8 2的比例随机分为 1 560张图像(训练集)和 390张图像(测试集)。3.2评价指标本次实验主要研究的是工业生产环境下的表面缺陷,既要满足实时性又要获得较好的检测性能,因此,采用检测速度 FPS(Frames Per Second)与平均精度均值(mean Average Precision,mAP)作为评价指标来比较各算法。FPS 是每秒可检测完的图片数量,mAP 是各类缺陷检测精度的平均值。mAP 的计算公式如式(2)所示:mmAP=i=0nAAP()in(2)其中:n为缺陷类别数;AAP为每类缺陷的平均精度值,可由 P-
33、R曲线下的阴影面积获得。3.3实验结果本 次 实 验 是 在 ubuntu20.04 操 作 系 统 下 基 于PyTorch深度学习框架实现的,GPU选用显存大小为24 GB 的 NVIDIA GeForce RTX 3090,CPU 配 置 是8 核 的 AMD EPYC 7601,CUDNN 版 本 为 11.1,PyTorch版本是 1.8.0,Python语言环境是 3.8.10。实验的超参数配置如下:在模型训练阶段,参数调整使用 SGD 优化器,初始学习率设为 0.01,动量为 0.937,使用 warm up 学习率衰减,权重衰减系数设置为 0.000 5。此外,批量大小设置为
34、1,总共训练 150 个轮数。本文算法检测得到的 P-R 曲线如图 7 所示。P-R 曲线的横轴为召回率,纵轴为精确率。由图 7可知,各类 P-R曲线下的面积都很大,AP值较高,即模型的检测性能较好。其中,pit缺陷的平均精度值达到0.986,scratch缺陷的平均精度值达到0.966。3.4对比分析与消融实验3.4.1聚类算法对比本 节 通 过 聚 类 算 法 的 对 比 实 验,验 证 使 用K-means+聚类算法生成的先验框对液压阀块表面微小缺陷检测的有效性。在相同训练参数的前提下,将K-means、K-means+算法在实验数据集上聚类生成的先验框分别应用在原YOLOv7算法上,对
35、比结果如表1所示。图 7本文算法的 P-R曲线Fig.7P-R curves of the algorithm in this paper表 12种先验框的平均精度值 Table 1Average precision values of two kinds of prior boxes%算法YOLOv7YOLOv7先验框K-means+聚类所得的先验框K-means聚类所得的先验框AP值92.789.2图 6聚类结果Fig.6Cluster results3072023年 11月 15日Computer Engineering 计算机工程从表 1可知,使用 K-means+聚类算法聚类生成的
36、先 验 框 应 用 在 原 YOLOv7 上 时 平 均 精 度 值 比K-means 聚类算法高 3.5 个百分点,K-means+聚类可提高边框回归位置预测的精确度。3.4.2注意力机制对比在多尺度特征融合模块后引入 CBAM、SE 以及CA 这 3 种不同的注意力机制,在 K-means+算法的基础上对网络进行训练,记录得到的平均精度值,对比结果如表 2所示。从表 2 可以看出,引入 SE 以及 CA 注意力机制都可以提高平均精度值,但引入 CBAM 注意力机制后平均精度值反而降低。引入 CA 注意力机制后平均精度值相较于 CBAM 与 SE 分别提高 2.6 与 0.8 个百分点。综上
37、所述,引入 CA注意力机制对提高小目标检测精度具有有效性。3.4.3YOLOv7算法与本文算法的对比从数据集中分别抽出 4 张图片进行对比,标注出真实框的 4 张原图如图 8 所示。使用 YOLOv7 与本文算法对这 4 张图片进行检测,将检测到的缺陷部分截取出来进行对比,结果如图 9所示。由图 9可知,YOLOv7 算法相对于本文算法更容易出现漏检的情况,且部分缺陷无法被识别或者识别出的置信度值较低,而本文算法能够尽可能准确地检测出图上存在的微小缺陷。表 2引入不同注意力机制的平均精度值 Table 2Average precision values of introducing diffe
38、rent attention mechanisms%注意力机制CBAMSECAAP值91.593.394.1图 8标注出真实框的原图Fig.8Original drawing marked with real frame图 92种算法的检测结果对比Fig.9Comparison of detection results of two algorithms308第 49卷 第 11期季娟娟,王佳,陈亚杰,等:基于改进 YOLOv7的液压阀块表面微小缺陷检测3.4.4不同算法的对比选 择 本 文 改 进 的 YOLOv7 算 法 与 Faster R-CNN、YOLOv4、MobileNetv3-
39、YOLOv4、YOLOv5-X、YOLOv7、YOLOv7-W6、YOLOv7-E6E 这 7种检测算法进行对比实验,将测试时的参数量、精确率、召回率、pit 精度值、scratch 精度值、平均精度值以及检 测 速 度 作 为 评 估 指 标,对 比 结 果 如 表 3 所 示。由表 3 可知,改进后的 YOLOv7 算法在所有算法中检 测 精 度 最 高,比 Faster R-CNN、YOLOv4、YOLOv5-X 以及 MobileNetv3-YOLOv4 这 4 种算法的平均精度值至少高出 45 个百分点,可见这 4 种算 法 对 本 文 数 据 集 中 的 微 小 缺 陷 检 测 能
40、 力 非 常差,漏检率高。虽然 YOLOv7-W6 与 YOLOv7-E6E的检测精度相较于原 YOLOv7 算法高,但是参数量非常大,导致检测速度很低,不符合实际工业应用场景的要求。本文算法检测速度高于 YOLOv7-W6 与 YOLOv7-E6E,平均精度值比两者分别高出4.9 和 1.8 个百分点。综上所述,本文算法能够在保证高实时性的前提下准确检测液压阀块表面的微小缺陷。3.4.5消融实验在使用相同训练参数的前提下,对改进模块进行消融实验。逐步加入 K-means+锚框聚类算法、CA注意力机制、改进的 UpC多支路上采样结构与改进的 ELAN-RepConv 结构,通过多个评价指标验证
41、每一步改进方法对检测精度提升的有效性。消融实验 结 果 如 表 4 所 示。由 表 4 可 知,通 过 添 加K-means+聚类算法,平均精度值提高 3.5 个百分点,引入 CA注意力机制后又提高 1.4个百分点,添加UpC多分支上采样结构,平均精度值进一步提高,最后使用 ELAN-RepConv 结构将平均精度值提高到97.6%。可知,与原始 YOLOv7 算法相比,改进算法使 液 压 阀 块 表 面 缺 陷 检 测 的 平 均 精 度 值 提 高 了8.4个百分点。4结束语 本文针对液压阀块表面缺陷尺寸微小、背景复杂、对比度低的特点,提出一种基于改进 YOLOv7的液压阀块表面微小缺陷检
42、测算法。通过引入 CA 注意力机制,同时使用改进的 UpC 多支路上采样结构与改进的 ELAN-RepConv 结构,提高对微小缺陷的特征提取能力并减少特征信息丢失。同时,为了增强 本 文 算 法 的 鲁 棒 性,采 用 离 线 数 据 增 强 融 合Mosaic数据增强以及 K-means+锚框聚类算法来提高算法的性能。多个对比实验和消融实验结果表明,改进的 YOLOv7 目标检测算法平均精度值达到97.6%,比原 YOLOv7算法提高 8.4个百分点,检测速度达到 55.2 frame/s,符合实际生产需求。下一步将在模型的主干特征提取部分采用 Transformer网络对算法结构进行改进
43、,在提高检测速度的前提下进一步提升液压阀块表面微小缺陷检测精度。表 3不同算法的结果对比 Table 3Comparison of results of different algorithms算法Faster R-CNNYOLOv4MobileNetv3-YOLOv4YOLOv5-XYOLOv7YOLOv7-W6YOLOv7-E6E本文算法参数量/10664.014.086.736.580.9151.737.3精确率/%21.586.760.082.685.193.793.492.1召回率/%19.39.312.632.283.089.091.398.0pit精度值/%3.049.045.0
44、56.690.593.596.498.6scratch精度值/%24.015.014.048.588.091.995.396.6平均精度值/%13.532.029.552.589.292.795.897.6检测速度/(frame s-1)11.835.789.816.764.925.913.855.2表 4消融实验结果 Table 4Results of ablation experimentK-means+锚框聚类算法CA注意力机制UpC多支路上采样结构ELAN-RepConv结构参数量/10636.536.536.537.337.3AP值/%89.292.794.197.097.6检测速度
45、/(frame s-1)64.964.959.855.255.23092023年 11月 15日Computer Engineering 计算机工程参考文献 1 张超,周雷,赵聪,等.轻量节能型液压阀块设计与增材制造 J.机械工程学报,2021,57(24):123-131.ZHANG C,ZHOU L,ZHAO C,et al.Design and additive manufacturing of lightweight energy-saving hydraulic valve block J.Journal of Mechanical Engineering,2021,57(24):1
46、23-131.(in Chinese)2 MENG Z W,ZHANG H X,YANG J,et al.Research on matching of power transmission system of electro-hydraulic hybrid electric vehicle J.IOP Conference Series:Earth and Environmental Science,2021,632(3):032007.3 YANG K F,WANG Y M,FU K X.Design of hydraulic bulging die for automobile tor
47、sion beam and optimization of forming process parameters EB/OL.2022-12-05.https:/ LIU Z Q,JIN B,DONG J K,et al.Design and control of a hydraulic hexapod robot with a two-stage supply pressure hydraulic system J.Machines,2022,10(5):305.5 ZHU Q,HUANG D,YU B,et al.An improved method combined SMC and ML
48、ESO for impedance control of legged robots electro-hydraulic servo system J.ISA Transactions,2022,130:598-609.6 MAR J C.Aerospace actuators M.S.l.:John Wiley&Sons,Inc.,2016.7 HU P,ZHU J X,GONG J,et al.Development of a comprehensive driving cycle for construction machinery used for energy recovery sy
49、stem evaluation:a case study of medium hydraulic excavators EB/OL.2022-12-05.https:/ YANG J,LIU B,ZHANG T,et al.Application of energy conversion and integration technologies based on electro-hydraulic hybrid power systems:a review J.Energy Conversion and Management,2022,272:116372.9 BENI J,KARLUI J,
50、ITUM,et al.Direct driven hydraulic system for skidders J.Energies,2022,15(7):2321.10 WANG J M,QIAO J P,GUO M C.Research on bearing surface defect detection system based on machine vision J.Journal of Physics:Conference Series,2022,2290(1):012061.11 WU Y,LIU J,ZHANG Y Q,et al.Detection algorithm of a
©2010-2024 宁波自信网络信息技术有限公司 版权所有
客服电话:4008-655-100 投诉/维权电话:4009-655-100