首页 | 本学科首页   官方微博 | 高级检索  
     

基于边缘AI计算的轨道扣件状态定量化检测方法
作者姓名:白堂博  段嘉明  杨建伟  许贵阳
作者单位:1.北京建筑大学机电与车辆工程学院;2.北京建筑大学城市轨道交通车辆服役性能保障北京市重点实验室
基金项目:国家自然科学基金资助项目(52272385);北京市自然科学基金资助项目(L211007,L221027);
摘    要:针对轨道扣件车载实时定量检测中检测的速度与精度难以平衡以及设备空间有限的问题,提出一种基于边缘AI计算的改进YOLOv8(You Only Look Once version 8)轨道扣件定量检测方法,并进行了部署和测试。首先,为实现轻量化部署,优化YOLOv8网络结构,嵌入移动神经网络V3(MobileNetV3)轻量化网络,引入压缩激励(SE)注意力机制,并重构颈部网络,加入可变形卷积;其次,结合推理引擎加速方法优化重构的网络模型,并将其部署在Jetson AGX Xavier边缘AI计算设备上;最后,对分割结果进行像素级提取,引入最小外接矩阵,定量分析扣件的断裂程度和偏转角度,并优化检测结果。结果表明:改进后的轻量化网络参数数量减少了22%;在边缘AI计算设备上的帧率相对原YOLOv8模型提升了80%,达到58帧·s-1;通过定量分析结果对模型进行修正,平均精度达到了97.0%,满足检测车辆所需的最低45帧·s-1检测要求,实现了轨道扣件的定量化实时检测。

关 键 词:边缘AI计算  扣件  轻量化  定量化检测  YOLOv8  
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号