钟佩思, 毕研治, 王晓, 徐杨, 刘梅. 改进YOLOv7算法的堆叠工件识别[J]. 制造技术与机床, 2024, (10): 145-150. DOI: 10.19287/j.mtmt.1005-2402.2024.10.020
引用本文: 钟佩思, 毕研治, 王晓, 徐杨, 刘梅. 改进YOLOv7算法的堆叠工件识别[J]. 制造技术与机床, 2024, (10): 145-150. DOI: 10.19287/j.mtmt.1005-2402.2024.10.020
ZHONG Peisi, BI Yanzhi, WANG Xiao, XU Yang, LIU Mei. Stacked workpiece recognition based on improved YOLOv7 algorithm[J]. Manufacturing Technology & Machine Tool, 2024, (10): 145-150. DOI: 10.19287/j.mtmt.1005-2402.2024.10.020
Citation: ZHONG Peisi, BI Yanzhi, WANG Xiao, XU Yang, LIU Mei. Stacked workpiece recognition based on improved YOLOv7 algorithm[J]. Manufacturing Technology & Machine Tool, 2024, (10): 145-150. DOI: 10.19287/j.mtmt.1005-2402.2024.10.020

改进YOLOv7算法的堆叠工件识别

Stacked workpiece recognition based on improved YOLOv7 algorithm

  • 摘要: 针对实际生产场景中堆叠遮挡的工件存在漏检率较高与实时性较差的问题,提出了一种改进的YOLOv7轻量化的目标检测方法。在主干网络中获取的3个特征层分别融入ECA注意力机制,能更好地注意到被遮挡的工件信息,降低漏检率;将主干网络和头部网络的普通卷积替换成深度可分离卷积,减少网络模型的参数量,提高检测速度;提出了一种联系周边特征信息的ELAN-DE模块,引入深度可分离卷积并在后面加入ECA注意力机制,提高了系统的稳定性。试验结果表明改进的YOLOv7算法相较于原始算法,在提高了近5.52%的平均精度(mAP)的情况下,模型大小减小了近21%,检测速度提升了18.7 f/s,检测效果也优于其他经典目标检测网络算法,满足了实际场景中的工件识别需求。

     

    Abstract: Aiming at the problem of high missed detection rate and poor real-time performance of stacked occluded workpieces in actual production scenarios, an improved YOLOv7 lightweight target detection method is proposed. The three feature layers obtained in the backbone network are integrated into the ECA attention mechanism respectively, which can better notice the occluded workpiece information and reduce the missed detection rate. The ordinary convolution of the backbone network and the head network is replaced by a deep separable convolution to reduce the number of parameters of the network model and improve the detection speed. An ELAN-DE module is proposed, which is associated with peripheral feature information. The depth separable convolution is introduced and the ECA attention mechanism is added to improve the stability of the system. The experimental results show that compared with the original algorithm, the improved YOLOv7 algorithm improves the average accuracy (mAP) by nearly 5.52%, the model size is reduced by nearly 21%, the detection speed is increased by 18.7 f/s, and the detection effect is also better than other classical target detection network algorithms, which meets the workpiece recognition requirements in the actual scene.

     

/

返回文章
返回