张明路,郭 策,吕晓玲,张 艳.改进的轻量化 YOLOv4 用于电子元器件检测[J].电子测量与仪器学报,2021,35(10):17-23
改进的轻量化 YOLOv4 用于电子元器件检测
Improved lightweight YOLOv4 for electronic components detection
  
DOI:
中文关键词:  目标检测  深度学习  可分离卷积  注意力机制  电子元器件
英文关键词:object detection  deep learning  separable convolution  attention mechanism  electronic components
基金项目:国家重点研发计划(2017YFB1303701)、国家重点自然基金(61733001)项目资助
作者单位
张明路 1.河北工业大学 机械工程学院 
郭 策 1.河北工业大学 机械工程学院 
吕晓玲 1.河北工业大学 机械工程学院 
张 艳 1.河北工业大学 机械工程学院 
AuthorInstitution
Zhang Minglu 1.School of Mechanical Engineering, Hebei University of Technology 
Guo Ce 1.School of Mechanical Engineering, Hebei University of Technology 
Lv Xiaoling 1.School of Mechanical Engineering, Hebei University of Technology 
Zhang Yan 1.School of Mechanical Engineering, Hebei University of Technology 
摘要点击次数: 1081
全文下载次数: 1878
中文摘要:
      针对电子行业制造机器人对电子元器件检测精度低和速度慢的问题,提出基于改进 YOLOv4 的电子元器件检测方法。 对网络结构进行改进,利用深度可分离卷积代替 PAN 网络中的传统卷积,提高检测速度;利用一种具有线性瓶颈的逆残差结构 代替 CSP darknet53 主干网络,降低模型参数,进一步提高检测效率;在检测网络 YOLO head 前添加注意力机制,提高检测精度。 模拟工业传送带环境建立了电子元器件数据集并进行数据增强,相较于原算法,精度( mAP) 提高了 1. 31%,速度提高了 16. 34 fps,权重大小从 245 下降到 41. 20 MB。 研究可为相关电子行业制造机器人的研制提供技术参考。
英文摘要:
      Aiming at the problem of low accuracy and slow speed of electronic components detection by manufacturing robots in the electronics industry, an electronic component detection method based on improved YOLOv4 is proposed. The network structure was improved by using depth-separable convolution instead of the traditional convolution in PAN networks to improve the detection speed. An inverse residual structure with a linear bottleneck was used instead of the CSP darknet53 backbone network to reduce the model parameters and further improve the detection efficiency. An attention mechanism was added before the YOLO head of the detection network to improve the detection accuracy. A data set of electronic components was established to simulate the industrial environment with conveyor belt and the data was enhanced. Compared with the original algorithm, the accuracy (mAP) is increased by 1. 31%, the speed is increased by 16. 34 fps, and the weight size is reduced from 245 to 41. 20 MB. The research can provide technical reference for the development of manufacturing robots in the electronics industry.
查看全文  查看/发表评论  下载PDF阅读器