现代制造工程 ›› 2022, Vol. 506 ›› Issue (11): 108-114.doi: 10.16731/j.cnki.1671-3133.2022.11.018

• 仪器仪表/检测/监控 • 上一篇    

基于改进YOLOv5的移动端螺栓缺失检测方法

陈欣瑞,周洋,赵屹涛,闫宪峰   

  • 发布日期:2023-01-06
  • 基金资助:
    河南省自然科学基金项目(202300410431)

  • Published:2023-01-06

摘要: 大型结构件中螺栓缺失往往会产生重大的安全隐患,而目前通过深度学习进行目标检测的设备通常部署繁琐、困难,为了保证检测的高效率并且足够轻量化的性能,提出一种通过改进YOLOv5进行模型训练,并将模型轻量化移植到移动端的方法,以实现螺栓的快速识别与个数缺失统计与报警。首先利用LabelImg软件对现有的螺栓数据进行标注,标注完成后对该数据集进行训练,再将得到的模型文件转化并降低位宽,转化为TensorFlow Lite(TFLite)模型文件,进而在安卓端进行部署。结果表明,将模型部署在小米MI 9 SE手机上,进行单目标检测时准确率可以达到98%,多目标检测时准确率可以达到97%,推理时间在40 ms左右,且报警功能可以正常使用,为螺栓缺失的检测提供了一种新的轻量化方法。

关键词: 目标检测; 深度学习; 改进YOLOv5; 轻量化; 螺栓连接

中图分类号: 


版权所有 © 《现代制造工程》编辑部 
地址:北京市东城区东四块玉南街28号 邮编:100061 电话:010-67126028 电子信箱:2645173083@qq.com
本系统由北京玛格泰克科技发展有限公司设计开发 技术支持:support@magtech.com.cn
访问总数:,当日访问:,当前在线: