科研之友
微信
新浪微博
Facebook
分享链接
摘要
为满足生产环境中钢材表面缺陷实时分类识别的需求,提出一种可部署于移动端的轻量卷积神经网络模型用于高效识别钢材表面缺陷。通过知识蒸馏将ResNet50和MobileNetV3分别作为教师模型和学生模型,生成改进的轻量神经网络模型MobileNetV3smalltp。利用MobileNetV3smalltp作为预训练模型,选用NEU(Northeastern University)带钢表面缺陷数据集进行微调,同时使用数据增强以模拟实际工况,得到模型在测试集中精度达到100%。最后将模型部署于手机上进行实际钢材表面缺陷识别。移动端模型测试、验证以及实际检测结果均显示MobileNetV3smalltp模型具有流畅的运行速度和较高的识别率,能够实现现场钢材表面缺陷的准确高效识别。
- 单位
