CN114283319A - 一种机车轮对踏面剥离识别方法 - Google Patents
一种机车轮对踏面剥离识别方法 Download PDFInfo
- Publication number
- CN114283319A CN114283319A CN202111600329.8A CN202111600329A CN114283319A CN 114283319 A CN114283319 A CN 114283319A CN 202111600329 A CN202111600329 A CN 202111600329A CN 114283319 A CN114283319 A CN 114283319A
- Authority
- CN
- China
- Prior art keywords
- frame
- sample image
- sample set
- image sample
- tread
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
Images
Landscapes
- Image Analysis (AREA)
Abstract
本发明公开了一种机车轮对踏面剥离识别方法,包括采集踏面剥离图像构建正样本图像样本集,采集未踏面剥离图像样本构建负样本图像样本集,对正样本图像样本集和负样本图像样本集进行人工标注获得真实框;利用聚类算法对真实框进行聚类处理,获得特征值;利用yolo神经网络,根据特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框自动筛选先验框,将含有剥离情况的先验框进行分类和回归预测,得到分类损失和位置损失;利用误差反向传播算法离线训练分类损失和位置损失进而优化神经网络中的权重参数。本发明基于yolo神经网络进行训练和检测,检测速度快,准确率高,仅通过图片和录像就能进行检测,极大的节约了资源和时间成本。
Description
技术领域
本发明涉及机车识别领域,尤其涉及一种机车轮对踏面剥离识别方法。
背景技术
现有的机车轮对踏面剥离识别方法通常利用光电技术非接触测量踏面缺陷的方法。用精密的激光位移传感器对踏面进行快速扫描,采集值经过处理后转换为数字图像,再通过传统的信号处理技术识别表面是否存在缺陷。这种识别方法需要精密激光位移传感器,造价高,应用场合限制。需要用到传统信号处理方法,会受到环境干扰,需要人工干预检测,识别速度慢。
发明内容
本发明提供一种机车轮对踏面剥离识别方法,以克服现有的剥离识别方法识别速度慢等技术问题。
为了实现上述目的,本发明的技术方案是:
一种机车轮对踏面剥离识别方法,包括以下步骤:
步骤1、筛选摄像头或录像机上采集的踏面剥离图像样本,保留满足设定分辨率的图像样本,构成正样本图像样本集;
步骤2、筛选摄像头或录像机上采集的未踏面剥离图像样本,保留满足设定分辨率的图像样本,构成负样本图像样本集;
步骤3、对正样本图像样本集和负样本图像样本集进行人工标注获得真实框;
步骤4、利用聚类算法对真实框进行聚类处理,获得特征值;
步骤5、利用yolo神经网络,根据所述特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框;
步骤6、通过yolo神经网络自动筛选先验框,将含有剥离情况的先验框进行分类和回归预测,得到分类损失和位置损失;
步骤7、利用误差反向传播算法离线训练分类损失和位置损失进而优化yolo神经网络中的权重参数;
步骤8、将权重参数保存为文本格式,通过分类器预测出剥离情况。
进一步的,所述步骤6中得到位置损失具体方法为:通过逻辑回归预测出先验框与真实框的偏移量,再通过交并补运算得到真实框与先验框之间的IOU值,即位置损失。
进一步的,计算IOU值的公式为:
其中,J(A,B)为真实框与先验框之间的IOU值,|A∩B|为真实框的IOU值,|A∪B|为先验框的IOU值。
进一步的,步骤4中具体为:利用聚类算法对真实框进行聚类处理,使真实框划分为不同网格,每个网格生成不同规格的先验框。
进一步的,步骤5中通过yolo神经网络卷积对正样本图像样本集和负样本图像样本集进行不同倍数的卷积采样,得到不同尺寸的先验框。
有益效果:本发明仅基于yolo模型进行训练和检测,检测速度快,准确率高,仅通过固定的摄像头或录像机就能进行检测,不需要精密的光学仪器,避免了环境中的噪声、热干扰、电磁波干扰等环境因素对预测结果产生的影响,同时摄像头和录像机能够自动捕捉发生剥离的区域,不需要人工干预,极大的节约了资源和时间成本。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作一简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。
图1为本发明机车轮对踏面剥离识别方法流程图。
具体实施方式
为使本发明实施例的目的、技术方案和优点更加清楚,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
本实施例提供了一种机车轮对踏面剥离识别方法,如图1,包括以下步骤:
一种机车轮对踏面剥离识别方法,其特征在于,包括以下步骤:
步骤1、筛选摄像头或录像机上采集的踏面剥离图像样本,保留满足设定分辨率的图像样本,构成正样本图像样本集;
步骤2、筛选摄像头或录像机上采集的未踏面剥离图像样本,保留满足设定分辨率的图像样本,构成负样本图像样本集;
步骤3、对正样本图像样本集和负样本图像样本集进行人工标注获得真实框;
步骤4、利用聚类算法对真实框进行聚类处理,获得不同的大小和尺寸值,即特征值;
步骤5、利用yolo神经网络,根据所述特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框;
步骤6、通过yolo神经网络自动筛选先验框,将含有剥离情况的先验框进行分类和回归预测,得到分类损失和位置损失;
步骤7、利用误差反向传播算法离线训练分类损失和位置损失进而优化yolo神经网络中的权重参数;
步骤8、将权重参数保存为文本格式,通过分类器预测出剥离情况。
在具体实施例中,所述步骤6中得到位置损失具体方法为:通过逻辑回归预测出先验框与真实框的偏移量,再通过交并补运算得到真实框与先验框之间的IOU值,即位置损失。
在具体实施例中,计算IOU值的公式为:
其中,J(A,B)为真实框与先验框之间的IOU值,|A∩B|为真实框的IOU值,|A∪B|为先验框的IOU值。
在具体实施例中,步骤4中具体为:利用聚类算法对真实框进行聚类处理,使真实框划分为不同网格,每个网格生成不同规格的先验框。首先将得到的真实框划分为13*13个网格,每个网格会生成长宽不同,大小不同的9个先验框。先验框的大小由K-Means聚类算法对所有预测框的长宽聚类得到;每个先验框都会预测三类值:1.位置信息包括中心坐标tx和ty,框的宽和长;2.有无目标预测;3.所属正负样本的剥离情况;
在具体实施例中,步骤5中通过yolo神经网络卷积对正样本图像样本集和负样本图像样本集进行不同倍数的卷积采样,得到不同尺寸的先验框;具体的,步骤5中通过多层卷积神经网络得到32倍,16倍,8倍三次不同大小的采样,获得不同尺寸先验框,每个先验框所对应的感受野不同,这样对所获得的不规整数据集做目标检测十分有利,可以确定数据中不同大小的目标种类。
在实际剥离过程中,只有第一次采集的正样本图像样本集和负样本图像样本集需要人工标注,由于人工标注已经训练好yolo神经网络,所以后续再进行预测无需人工标注,采集正样本图像样本集和负样本图像样本集后即可识别剥离情况。
项目 | YOLO-v3 |
准确率 | 91.5% |
精确率 | 89.52% |
召回率 | 94% |
假正率 | 11% |
真正率 | 94% |
F-measure | 91.69% |
表1:剥离识别准确率统计表
表1为应用本发明的剥离识别方法获得的识别准确率统计表,由表1可知应用本申请后,剥离识别准确率和真正率高,对发生剥离处的识别效果良好。
最后应说明的是:以上各实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述各实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分或者全部技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的范围。
Claims (5)
1.一种机车轮对踏面剥离识别方法,其特征在于,包括以下步骤:
步骤1、筛选摄像头或录像机上采集的踏面剥离图像样本,保留满足设定分辨率的图像样本,构成正样本图像样本集;
步骤2、筛选摄像头或录像机上采集的未踏面剥离图像样本,保留满足设定分辨率的图像样本,构成负样本图像样本集;
步骤3、对正样本图像样本集和负样本图像样本集进行人工标注获得真实框;
步骤4、利用聚类算法对真实框进行聚类处理,获得特征值;
步骤5、利用yolo神经网络,根据所述特征值将正样本图像样本集和负样本图像样本集进行特征提取获得先验框;
步骤6、通过yolo神经网络自动筛选先验框,将含有剥离情况的先验框进行分类和回归预测,得到分类损失和位置损失;
步骤7、利用误差反向传播算法离线训练分类损失和位置损失进而优化yolo神经网络中的权重参数;
步骤8、将权重参数保存为文本格式,通过分类器预测出剥离情况。
2.如权利要求1所述的一种机车轮对踏面剥离识别方法,其特征在于,所述步骤6中得到位置损失具体方法为:通过逻辑回归预测出先验框与真实框的偏移量,再通过交并补运算得到真实框与先验框之间的IOU值,即位置损失。
4.如权利要求3所述的一种机车轮对踏面剥离识别方法,其特征在于,步骤4中具体为:利用聚类算法对真实框进行聚类处理,使真实框划分为不同网格,每个网格生成不同规格的先验框。
5.如权利要求4所述的一种机车轮对踏面剥离识别方法,其特征在于:步骤5中通过yolo神经网络卷积对正样本图像样本集和负样本图像样本集进行不同倍数的卷积采样,得到不同尺寸的先验框。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111600329.8A CN114283319A (zh) | 2021-12-24 | 2021-12-24 | 一种机车轮对踏面剥离识别方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111600329.8A CN114283319A (zh) | 2021-12-24 | 2021-12-24 | 一种机车轮对踏面剥离识别方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN114283319A true CN114283319A (zh) | 2022-04-05 |
Family
ID=80875099
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202111600329.8A Pending CN114283319A (zh) | 2021-12-24 | 2021-12-24 | 一种机车轮对踏面剥离识别方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114283319A (zh) |
Cited By (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116310656A (zh) * | 2023-05-11 | 2023-06-23 | 福瑞泰克智能***有限公司 | 训练样本确定方法、装置和计算机设备 |
-
2021
- 2021-12-24 CN CN202111600329.8A patent/CN114283319A/zh active Pending
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116310656A (zh) * | 2023-05-11 | 2023-06-23 | 福瑞泰克智能***有限公司 | 训练样本确定方法、装置和计算机设备 |
CN116310656B (zh) * | 2023-05-11 | 2023-08-15 | 福瑞泰克智能***有限公司 | 训练样本确定方法、装置和计算机设备 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111179249A (zh) | 一种基于深度卷积神经网络的电力设备检测方法和装置 | |
CN109472226B (zh) | 一种基于深度学习的睡觉行为检测方法 | |
CN113324864B (zh) | 一种基于深度学习目标检测的受电弓碳滑板磨耗检测方法 | |
CN117893817A (zh) | 晶圆缺陷分类方法及其装置、***、电子设备和存储介质 | |
CN111369526B (zh) | 基于半监督深度学习的多类型旧桥裂痕识别方法 | |
CN112365497A (zh) | 基于TridentNet和Cascade-RCNN结构的高速目标检测方法和*** | |
CN115631472B (zh) | 一种高速公路行人闯入智能检测方法 | |
CN110751619A (zh) | 一种绝缘子缺陷检测方法 | |
CN114897806A (zh) | 缺陷检测方法、电子设备及计算机可读存储介质 | |
CN114897789B (zh) | 一种基于图像分割的烧结矿粒度检测方法及*** | |
CN115311507A (zh) | 基于数据处理的建筑板材分类方法 | |
Shanthakumari et al. | Mask RCNN and Tesseract OCR for vehicle plate character recognition | |
CN114283319A (zh) | 一种机车轮对踏面剥离识别方法 | |
CN112183554B (zh) | 一种自动化道路边界轮廓提取方法 | |
CN111597939B (zh) | 一种基于深度学习的高铁线路鸟窝缺陷检测方法 | |
CN113283396A (zh) | 目标对象的类别检测方法、装置、计算机设备和存储介质 | |
CN116843677A (zh) | 钣金件的外观质量检测***及其方法 | |
CN116994161A (zh) | 一种基于改进YOLOv5的绝缘子缺陷检测方法 | |
CN110826564A (zh) | 一种复杂场景图像中的小目标语义分割方法及*** | |
CN116486296A (zh) | 目标检测方法、装置及计算机可读存储介质 | |
CN115909151A (zh) | 一种复杂工作条件下运动容器编号识别方法 | |
CN115761211A (zh) | 一种基于rgb图像与热成像耦合的石化企业泵房设备温度监测方法及装置 | |
CN115082449A (zh) | 一种电子元器件缺陷检测方法 | |
Roy | Automatics number plate recognition using convolution neural network | |
CN111523583A (zh) | 一种利用无人机对设备铭牌照片自动识别分类的方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |