基于改进YOLOv8n的轴承外圈表面缺陷检测方法

本发明基于改进YOLOv8n的轴承外圈表面缺陷检测方法属于目标检测领域；该方法依次执行以下步骤：步骤a、采集轴承外圈表面缺陷图像，所述缺陷包括凹槽、擦伤、划痕和污渍；步骤b、对所述轴承外圈表面缺陷图像进行预处理，并构建成完整的数据集；步骤c、改进YOLOv8n模型，引入注意力模块、采样模块和特征提取融合模块；步骤d、训练改进的YOLOv8n模型；步骤e、输出改进YOLOv8n最优模型；步骤f、测试改进YOLOv8n最优模型；实验结果表明：通过引入注意力机制，下采样模块以及特征提取融合模块，在提高检测精度的同时，降低了模型的复杂度，提高了运算速度，同时，还改善了模型识别复杂缺陷的能力。