一种基于Faster-RCNN的轴承滚子瑕疵检测方法

摘要

一种基于Faster‑RCNN的轴承滚子瑕疵检测方法，包括以下步骤：1)获取数据，通过实验装置设备拍摄轴承滚子照片；2)将数据集分为训练集和预测集，为训练集中的数据制作单一缺陷标签；3)用训练集训练Faster‑RCNN网络；4)用训练好的Faster‑RCNN模型对预测集的图片进行检测，得到每张图片的检测结果；5)删除原有标签，为训练集中的数据制作15类缺陷标签，预测集中的数据不作处理；6)用新的训练集训练Faster‑RCNN网络；7)用新的Faster‑RCNN模型对预测集的图片进行检测，得到每张图片的检测结果。本发明构建Faster‑RCNN模型，准确率较高、检测速度较快。