一种线束连接器目标检测方法

本发明公开一种线束连接器目标检测方法，属于汽车线束生产领域。该方法包括以下步骤：步骤1、对线束连接器进行数据采集，得到原始图像数据集；然后对原始图像数据集进行处理，得到线束连接器数据集；步骤2、首先，在YOLOv5模型的主干部分引入C2f模块替换C3模块；其次，引入双向金字塔特征网络模块；步骤3、引入Focal‑EIoU损失函数作为新的损失函数；使用步骤1建立好的线束连接器数据集对步骤2建立的YOLO图像识别模型进行训练，通过训练后的模型完成对线束连接器的精准快速识别。本发明相较于其他的目标检测方法如双阶段检测方法的RNN系列算法和单阶段检测算法的SSD算法等，具有精度更高，实时性更强的特点。