摘要:针对电子行业制造机器人对电子元器件检测精度低和速度慢的问题,提出基于改进 YOLOv4 的电子元器件检测方法。 对网络结构进行改进,利用深度可分离卷积代替 PAN 网络中的传统卷积,提高检测速度;利用一种具有线性瓶颈的逆残差结构 代替 CSP darknet53 主干网络,降低模型参数,进一步提高检测效率;在检测网络 YOLO head 前添加注意力机制,提高检测精度。 模拟工业传送带环境建立了电子元器件数据集并进行数据增强,相较于原算法,精度( mAP) 提高了 1. 31%,速度提高了 16. 34 fps,权重大小从 245 下降到 41. 20 MB。 研究可为相关电子行业制造机器人的研制提供技术参考。