公开(公告)号：CN116402785A

公开(公告)日：2023-07-07

申请号：CN202310344743.X

申请日：2023-04-03

Applicant: 重庆科技学院

Inventor： 陈国荣 ,  钟银凤 ,  简夜明 ,  黄家铭 ,  袁巧 ,  李朗 ,  肖彦冰 ,  汪博成 ,  文婷婷 ,  张毅轩

IPC: G06T7/00 ,  G06T7/136 ,  G06T5/00 ,  G06T7/11 ,  G06T5/20 ,  G06V10/80 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明公开了一种基于视觉注意力机制的钢轨表面缺陷检测方法及系统方法包括如下步骤：S1：获取钢轨表面图像数据；S2：对获取的钢轨表面图像数据进行FT显著图提取；S3：对获取的钢轨表面图像数据进行CA显著图提取；S4：将步骤S2提取的FT显著图与步骤S3提取的CA显著图进行小波融合；S5：对融合后的显著图进行阈值分割；S6：使用YOLOv5算法架构对分割后的图像进行缺陷分类检测。其效果是：本发明可以克服光照影响，实现对钢轨的缺陷进行精准分类检测，为设备维护人员的缺陷探测带来便利。