一种基于深度学习的电子元件目标提取方法及系统

    公开(公告)号:CN117351224A

    公开(公告)日:2024-01-05

    申请号:CN202311316098.7

    申请日:2023-10-11

    Abstract: 本发明提供一种基于深度学习的电子元件目标提取方法及系统,采用多注意力机制增强型YOLO网络,在YOLOX骨干部分引入SE、CBAM注意力机制,在瓶颈结构后加入SE注意力机制,在三个有效输出特征层加入CBAM注意力机制;解耦预测部分,采用CIoU损失函数训练网络。本发明引入两种注意力机制,结合卷积神经网络与注意力机制的优势,集中计算能力在有效特征信息上,减少全连接层冗余信息,从空间和通道维度更好地聚焦小目标,使网络更加关注电路板上的微小元件,减少背景的影响;CIoU损失函数综合考虑预测框与真实框的重叠区域、中心点距离和长宽比,提高收敛速度,使预测框更贴近真实的电子元件目标,使定位框更加精确,提高检测精度。

Patent Agency Ranking