一种基于YOLOv8的螺栓紧固件识别方法

本发明公开了一种基于YOLOv8的螺栓紧固件识别方法，包括数据预处理：对螺栓图像进行尺寸调整、归一化处理及数据增强操作；输入图像特征分析：将预处理后的图像输入至特征聚焦模块和多尺度特征重建模块进行特征提取；特征融合：将特征聚焦模块的输出特征图与多尺度特征重建模块的输出特征图相加，得到融合特征；目标检测：将融合特征输入至改进的YOLOv8模型，输出螺栓紧固件的类别标签、边界框坐标及置信度分数；实时检测：基于训练完成的YOLOv8模型对输入图像进行实时识别。解决了现有技术中螺栓紧固件识别效率低、准确性差的问题，通过对螺栓图像数据进行训练，识别不同类型的螺栓紧固件，并实现实时检测，具有高效、准确、鲁棒的优点。