(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210492774.5 (22)申请日 2022.05.07 (71)申请人 无锡雪浪数制科技有限公司 地址 214000 江苏省无锡市经济开发区金 融一街1号昌兴国际金融大厦6楼 (72)发明人 顾毅　李淑霞　 (74)专利代理 机构 无锡华源专利商标事务所 (普通合伙) 32228 专利代理师 过顾佳 (51)Int.Cl. G06V 10/82(2022.01) G06V 10/764(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01) G06V 10/774(2022.01)G06K 9/62(2022.01) G06V 10/44(2022.01) F16B 39/24(2006.01) (54)发明名称 轻量化高铁轮毂垫片防松 识别方法及装置 (57)摘要 本发明涉及一种轻量化高铁轮毂垫片防松 识别方法及装置。 其包括： 构建基于YOL OX的垫片 防松目标识别模 型； 获取包含防松垫片的高铁轮 毂图像， 利用所构建的垫片防松目标识别模型对 所获取的高铁轮毂图像进行目标识别， 以输出垫 片防松检测框信息； 所述垫片防松 检测框信息包 括垫片贴合类别以及类别置信度， 当所述类别置 信度与置信度阈值匹配时， 则将所述垫片贴合类 别确定为防松垫片的当前贴合状态。 本发明能有 效实现对高铁轮毂防松垫片与轮毂螺栓间的贴 紧状态识别， 提高对高铁轮毂运行的安全检测智 能化程度与可靠性。 权利要求书2页 说明书7页 附图5页 CN 114842315 A 2022.08.02 CN 114842315 A 1.一种轻量 化高铁轮毂垫片防松 识别方法， 其特 征是， 所述垫片防松 识别方法包括： 构建基于 YOLOX的垫片防松目标识别模型； 获取包含防松垫片的高铁轮毂图像， 利用所构建的垫片防松目标识别模型对所获取的 高铁轮毂图像进行目标识别， 以输出垫片防松检测框信息； 所述垫片防松检测框信 息包括垫片贴合类别以及类别置信度， 当所述类别置信度与置 信度阈值匹配时， 则将所述垫片贴合类别确定为防松垫片的当前贴合状态。 2.根据权利要求1所述的轻量化高铁轮毂垫片防松识别方法， 其特征是： 构建垫片防松 目标识别模型中， 采用Shuf fleNetV2网络架构作为YOLOX的骨干网络 。 3.根据权利要求2所述的轻量化高铁轮毂垫片防松识别方法， 其特征是： 在 ShuffleNetV2网络架构的架构基础模块内配置一参数个数为k的ECA通道注意力模块， 其 中， k≤9； 架构基础模块工作时， 通过ECA通道注意力模块对架构基础模块内的卷积单元卷积计 算后的信息处理， 架构基础 模块内的卷积单元包括基础内1 ×1卷积模块以及基础内3 ×3卷 积模块， 基础内1 ×1卷积模块、 基础内3 ×3卷积模块与E CA通道注意力模块依次适配连接 。 4.根据权利 要求2所述的轻量化 高铁轮毂垫片防松识别方法， 其特征是： ShuffleNetV2 网络架构， 还包括降采样模块， 所述降采样模块包括采样第一分支、 采样第二分支以及与采 样第一分支、 采样第二分支适配连接的降采样连接模块， 其中， 采样第一分支包括采样第一分支3 ×3卷积模块以及采样第 一分支1×1卷积模块， 采样 第一分支3 ×3卷积模块 通过采样第一分支1 ×1卷积模块与降采样连接模块 适配连接； 卷积第二分支包括采样第二分支1 ×1卷积模块以及采样第 二分支3×3卷积模块， 采样 第二分支1 ×1卷积模块 通过采样第二分支3 ×3卷积模块降采样连接模块 适配连接 。 5.根据权利要求1所述的轻量化高铁轮毂垫片防松识别方法， 其特征是： 构建的垫片防 松目标识别模型中， 检测头内包括 40*40目标预测层以及80 *80的目标 预测层。 6.根据权利要求1至5任一项所述的轻量化高铁轮毂垫片防松识别方法， 其特征是， 构 建垫片防松目标识别模型时， 包括如下步骤： 步骤1、 提供基于 YOLOX的目标识别基础模型； 步骤2、 制作模型基础数据集， 所述模型基础数据集的模型基础数据至少包含高铁轮 毂、 轮毂锁紧螺栓以及与所述轮毂锁紧螺栓对应的防松垫片， 对模型基础数据内防松垫片 的垫片位置以及类别进行 标注， 所述标注的类别包括贴紧垫片或未贴合垫片； 模型基础数据集内同时包含标注贴紧垫片的模型基础数据以及标注未贴合垫片的模 型基础数据； 步骤3、 将上述制作的模型基础数据集划分为模型训练数据集与模型验证数据集， 利用 模型训练数据集对目标识别基础模型进行训练， 利用模型验证数据集对训练后的目标识别 基础模型进行验证； 在满足目标识别基础模型的终止条件后， 构建得到 垫片防松目标识别模型。 7.根据权利要求1至5任一项所述的轻量化高铁轮毂垫片防松识别方法， 其特征是， 垫 片防松目标识别模型同时输出多个垫片 检测预测框时， 对所有的垫片 检测预测框进行去重 处理， 以在去重后得到所需的垫片防松检测框信息 。 8.一种轻量化高铁轮毂垫片防松识别装置， 其特征是， 用于执行上述权利要求1～权利权　利　要　求　书 1/2 页 2 CN 114842315 A 2要求7任一项的高铁轮毂垫片防松 识别方法。 9.根据权利要求8所述轻量 化高铁轮毂垫片防松 识别装置， 其特 征是， 包括 图像采样模块， 用于获取待识别的高铁轮毂图像， 其中， 所获取的高铁轮毂图像包含防 松垫片； 识别处理模块， 接收图像获取模块获取的高铁轮毂图像， 并对所获取的高铁轮毂图像 利用垫片防松目标识别模型进行目标识别， 以输出垫片防松检测框信息； 识别输出模块， 将垫片防松检测框信息内的类别置信度与置信度阈值比较， 当所述类 别置信度与置信度阈值匹配时， 则将所述垫片贴合类别确定并输出为防松垫片的当前贴合 状态。 10.根据权利要求9所述轻量化高铁轮毂垫片防松识别装置， 其特征是， 所述图像采样 模块包括摄 像头。权　利　要　求　书 2/2 页 3 CN 114842315 A 3