一种基于视频监控的废钢车厢等级判定方法技术

本发明专利技术公开了一种基于视频监控的废钢车厢等级判定方法，属于废钢等级判定技术领域。该方法将收集的废钢图像集分成训练集和验证集，使用训练集训练深度学习网络模型，并通过验证集进行验证，当输出的废钢图像的类别与位置与验证集的标注进行比较，准确率均超过80％时，完成对深度学习网络模型的训练，随后基于视频监控实时采集废钢图像，输入训练好的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，计算各类别的废钢占比，根据贝叶斯统计方法结合先验知识，得出对一废钢车厢的每个评定等级的概率值，将概率值最高的评定等级作为该废钢车厢的评定等级。本发明专利技术的废钢车厢判定方法具有准确率高、避免人为干扰的特点。

【技术实现步骤摘要】
一种基于视频监控的废钢车厢等级判定方法
本专利技术涉及废钢等级判定
，具体地涉及一种基于视频监控的废钢车厢等级判定方法。
技术介绍
随着钢材需求的日益增长，废钢的回收再利用显得越来越重要。废钢，指的是钢铁厂生产过程中不成为产品的钢铁废料(如切边、切头等)以及使用后报废的设备、构件中的钢铁材料，成分为钢的叫废钢；成分为生铁的叫废铁，统称废钢。目前世界每年产生的废钢总量为3～4亿吨，约占钢总产量的45～50％，其中85～90％用作炼钢原料，10～15％用于铸造、炼铁和再生钢材。目前冶金企业对废钢铁质量检验有两大难题，一是扣杂质，二是废钢铁定级问题。质量检验主要对废钢铁场地实行封闭式定理，安装监控设施，严禁废钢供应商进入卸车现场，以免干扰质量检验人员的工作。要求废钢铁供应商分类进货，就是把重型、中型、小型、统料废钢分别装车，单独进货，这样就可以解决定级不准确的问题。为了更好地利用废钢，需要对废钢材料进行等级判定。现阶段，针对废钢等级判定方法等工作集中于指定车厢，针对整车厢废料进行粗略的等级判定等工作。基于人工的废钢等级判定方法主观性强，且只能基于车厢废料表面进行检测，堆砌于内部的废钢材料不能进行检测。该方法难以确保废钢材料等级判定的准确性，为后续的钢材质量加工等工作带来了不可控制的因素，钢坯质量也会受到一定程度的影响。
技术实现思路
针对现有技术中存在的问题，本专利技术提供了一种基于视频监控的废钢车厢等级判定方法。为实现上述目的，本专利技术提供如下技术方案：一种基于视频监控的废钢车厢等级判定方法，具体包括以下步骤：(1)收集废钢图像，根据先验知识对收集的废钢图像进行类别和位置标注，获得废钢图像集；所述废钢图像集包括训练集和验证集；(2)将训练集输入深度学习网络模型中进行训练，当损失函数值小于1且不再下降，停止训练，将验证集逐一输入经训练的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，当输出的废钢图像的类别与位置与验证集的标注进行比较，准确率均超过80％时，完成对深度学习网络模型的训练；(3)实时获取一废钢车厢卸钢的视频，将视频帧图像输入到训练好的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，计算各类别的废钢占比，根据贝叶斯统计方法结合先验知识，得出对一废钢车厢的每个评定等级的概率值，将概率值最高的评定等级作为该废钢车厢的评定等级。进一步地，所述深度学习网络模型包括：特征提取模块、分类模块、分割模块；所述特征提取模块的输出端分别与分类模块、分割模块的输入端连接；所述特征提取模块用于特征图的提取；所述分类模块用于预测废钢图像上的废钢类别与废钢位置；所述分割模块用于预测废钢图像的轮廓图像。进一步地，所述特征提取模块由主干网络和FPN网络连接而成。进一步地，步骤(2)中的训练过程具体为：将训练集中的一张废钢图像输入特征提取模块进行特征图像提取，提取出分辨率依次增大的四张特征图，将分辨率最大的特征图输入分割模块中进行目标掩膜分割，输出废钢图像的轮廓图像，将四张特征图输入分类模块中，输出废钢图像上的废钢类别与对应类别的废钢位置，将轮廓图像、废钢类别与对应类别的废钢位置进行融合，得到标有废钢类别和位置的废钢图像，依次将训练集中的废钢图像输入至深度学习网络模型中，重复上述过程，直至损失函数值小于1且不再下降，停止训练。进一步地，所述损失函数L为：L＝LH+LS其中，LH为复合损失函数，LS为二类交叉熵损失函数。进一步地，所述复合损失函数LH为：其中，N表示提取到的预测框总数，Lcls表示focal损失函数，Lreg表示交叉熵损失函数，(x,y)表示特征图中预测废钢的位置坐标，α为超参数，取值为1。进一步地，所述二类交叉熵损失函数LS为：LS＝BCE(M,Mgt)其中，M表示模型预测废钢类别的掩膜，Mgt表示废钢类别对应标注的掩膜。与现有技术相比，本专利技术具有如下有益效果：本专利技术的废钢车厢等级判定方法可以降低人为主观性的干扰，这对于提升钢材产品质量，保证产品生产的可靠性，提升钢材生产流程中的智能化检测水平，减少劳动人员强度会有很大贡献；同时基于深度学习的网络模型的废钢检测，能够在检测中再不断进行迭代，提升废钢车厢等级判定的准确度。附图说明图1为本专利技术所采用的深度学习网络模型结构图。具体实施方式下面将结合本专利技术的附图，对本专利技术中的技术方案进行清楚、完整地描述。基于本专利技术中的技术方案，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其它实施例，都属于本专利技术保护的范围。本专利技术基于视频监控的废钢车厢等级判定方法中对废钢的等级判定是基于深度学习网络进行的，可以通过对废钢材料类别和位置分析，准确分割出不同等级的废钢材料，为解决视觉传感器只能采集表面数据的问题，采取边采集边卸车的分层采集方式，也就是在废钢卸载出车厢时是一层一层的卸载的，可以对整个车厢内废钢进行实时监控并判定等级。本专利技术提供了一种基于视频监控的废钢车厢等级判定方法，该废钢车厢等级判定方法包括如下步骤：(1)收集废钢图像，根据先验知识对收集的废钢图像进行类别和位置标注，获得废钢图像集；废钢图像集包括训练集和验证集；(2)将训练集输入深度学习网络模型中进行训练，当损失函数值小于1且不再下降，停止训练，将验证集逐一输入经训练的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，当输出的废钢图像的类别与位置与验证集的标注进行比较，准确率均超过80％时，完成对深度学习网络模型的训练；如图1，本专利技术中所采用的深度学习网络模型包括：特征提取模块、分类模块、分割模块；所述特征提取模块的输出端分别与分类模块、分割模块的输入端连接。特征提取模块用于在输入的废钢图像上提取对应特征图，该特征图被分别送进分类和分割模块用于后续的检测，分类模块能够基于特征图预测出废钢图像上各废钢的分类置信度及对应分废钢位置，分割模块则基于特征图得到各单体目标的分割掩膜，预测废钢图像上的轮廓图像，分类模块和分割模块上的预测结果进行后处理融合后输出标有废钢种类和位置的废钢图像。步骤(2)具体包括如下子步骤：(2.1)将训练集中的一张废钢图像输入特征提取模块进行特征图像提取，提取出分辨率依次增大的四张特征图；(2.2)将分辨率最大的特征图输入分割模块中进行目标掩膜分割，输出废钢图像的轮廓图像；(2.3)将四张特征图输入分类模块中，输出废钢图像上的废钢类别与对应类别的废钢位置；(2.4)将步骤(2.2)输出的轮廓图像与步骤(2.3)输出的废钢类别及对应类别的废钢位置进行融合，得到标有废钢类别和位置的废钢图像；(2.5)依次将训练集中的废钢图像输入至深度学习网络模型中，重复步骤(2.1)-(2.4)，直至损失函数值小于1且不再下降，停止训练，深度学习网络模型迭代的次数过多，会降低深度神经网络模型的准确率。本步骤中所采用的损失函数L为：L＝LH+LS...

【技术保护点】
1.一种基于视频监控的废钢车厢等级判定方法，其特征在于，具体包括以下步骤：/n(1)收集废钢图像，根据先验知识对收集的废钢图像进行类别和位置标注，获得废钢图像集；所述废钢图像集包括训练集和验证集；/n(2)将训练集输入深度学习网络模型中进行训练，当损失函数值小于1且不再下降，停止训练，将验证集逐一输入经训练的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，当输出的废钢图像的类别与位置与验证集的标注进行比较，准确率均超过80％时，完成对深度学习网络模型的训练；/n(3)实时获取一废钢车厢卸钢的视频，将视频帧图像输入到训练好的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，计算各类别的废钢占比，根据贝叶斯统计方法结合先验知识，得出对一废钢车厢的每个评定等级的概率值，将概率值最高的评定等级作为该废钢车厢的评定等级。/n

【技术特征摘要】
1.一种基于视频监控的废钢车厢等级判定方法，其特征在于，具体包括以下步骤：
(1)收集废钢图像，根据先验知识对收集的废钢图像进行类别和位置标注，获得废钢图像集；所述废钢图像集包括训练集和验证集；
(2)将训练集输入深度学习网络模型中进行训练，当损失函数值小于1且不再下降，停止训练，将验证集逐一输入经训练的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，当输出的废钢图像的类别与位置与验证集的标注进行比较，准确率均超过80％时，完成对深度学习网络模型的训练；
(3)实时获取一废钢车厢卸钢的视频，将视频帧图像输入到训练好的深度学习网络模型中，输出标有废钢类别和位置的废钢图像，计算各类别的废钢占比，根据贝叶斯统计方法结合先验知识，得出对一废钢车厢的每个评定等级的概率值，将概率值最高的评定等级作为该废钢车厢的评定等级。


2.根据权利要求1所述基于视频监控的废钢车厢等级判定方法，其特征在于，所述深度学习网络模型包括：特征提取模块、分类模块、分割模块；所述特征提取模块的输出端分别与分类模块、分割模块的输入端连接；所述特征提取模块用于特征图的提取；所述分类模块用于预测废钢图像上的废钢类别与废钢位置；所述分割模块用于预测废钢图像的轮廓图像。


3.根据权利要求2所述基于视频监控的废钢车厢等级判定方法，其特征在于，所述特征提取模块由主干网络和FPN网络连接而成。


4.根...

【专利技术属性】
技术研发人员：李桂东，
申请(专利权)人：南京耘瞳科技有限公司，
类型：发明
国别省市：江苏;32