螺栓是电磁能装备中重要的连接紧固件,基于计算机视觉的检测有助于后续的螺栓松动检测、缺陷检测等。针对螺栓目标小的特征,提出了基于改进YOLOv5(you only look once)的螺栓目标检测方法。首先,对Backbone引入SE(squeeze-and-excitati...螺栓是电磁能装备中重要的连接紧固件,基于计算机视觉的检测有助于后续的螺栓松动检测、缺陷检测等。针对螺栓目标小的特征,提出了基于改进YOLOv5(you only look once)的螺栓目标检测方法。首先,对Backbone引入SE(squeeze-and-excitation)注意力机制模块,提高基础主干网络的特征提取能力;然后,引入Bi-FPN(bi-directional feature pyramid network)结构实现Neck部分的特征融合功能,加权融合不同层的特征信息;最后,建立由个人收集和公共数据集NPU-BOLD组成的螺栓数据集,共包含1520张图像。该数据集下的实验结果表明:该方法检测的mAP为90.0%,较YOLOv5s提升了2.4%,精度提升较明显且能够保持较快的检测速度。展开更多