摘要
本发明提供了一种螺栓表面缺陷检测方法及设备,包括:构建螺栓表面缺陷检测试验台;采集制造过程中的螺栓缺陷图像;对螺栓缺陷图像进行标注,对标注后的螺栓缺陷图像进行增强得到数据集,将数据集划分为训练集和测试集;构建光分组网络LightPGNet,定义双鉴别器条件空间‑语义特征融合架构,选用距离交并比DIoU替代广义交并比GIoU作为边界框回归的损失函数,根据迁移学习得到螺栓表面缺陷检测模型;采用训练集对螺栓表面缺陷检测模型进行训练;采用测试集对训练好的螺栓表面缺陷检测模型进行测试,并以精确率、召回率、平均精度均值、每秒帧数、类别平均精度及每秒十亿次浮点运算次数作为评价指标。