
点火线圈飞线检测,完成技术评估
2020 年 09 月 24 日,苏州元泰提供点火线圈原始图像,共计 1 类 803 张(不含重复图像),图像尺寸为 2448x2048。当日完成标注工作,并开始设计相关模型。
本项目包含 803 张图像,其中包含飞线特征共计 2054 处(部分图像中不止 1 处)。
图 1 含有多处飞线缺陷的图像
本项目大部分飞线缺陷都有清晰的轮廓,我们选用 YOLO 类模型可以执行缺陷定位和分类任务。
图 2 YOLO 模型的预测结果
a)人工标注的飞线缺陷
b)模型预测的飞线缺陷
图 3 模型的部分预测结果
结论
- 经过 200 个训练周期,模型的阳性检出率为94.2%,准确率为85.4%。
- 利用数据增强技术,扩充训练样本,提高模型的鲁棒性
- 目前的模型还有很多调整空间,尤其是需要更多的训练数据
相信经过一段时间的合作,我们一定可以获得更好的结果。