一种基于密集目标检测的钢筋计数处理方法技术

本发明专利技术公开了一种基于密集目标检测的钢筋计数处理方法，包括采集钢筋切面图片，确定先验框；输入钢筋切面图片，构建RPN网络模型生成包含预测框的候选区域；构建分支网路，预测候选区域中预测框的类别得分和相对坐标；构建损失函数，基于带动量的随机梯度下降法更新RPN网络模型参数；将更新的RPN网络模型迭代反馈，得到钢筋检测计数模型；对钢筋检测计数模型输出的预测框进行软性非极大值抑制处理，过滤掉重复预测框，计算得到钢筋切面图片中钢筋的数量。本发明专利技术设计科学合理，使用方便，解决了现有建筑工地中人力清点钢筋计数不准确的技术问题，具有较好的平均精度、召回率和准确率。具有较好的平均精度、召回率和准确率。具有较好的平均精度、召回率和准确率。

【技术实现步骤摘要】
一种基于密集目标检测的钢筋计数处理方法


[0001]本专利技术属于密集目标检测
，具体涉及一种基于密集目标检测的钢筋计数处理方法。

技术介绍

[0002]在建筑行业中，钢筋是不可缺少的建筑材料之一。在钢筋的生产、运输、销售等每个环节都必须精确计算钢筋根数，以及施工现场的工作人员需要对采购的钢筋进行清点和数量确认。传统的钢筋清点方式采用人工清点，该方式过程繁琐、耗时耗力且人力成本较高。
[0003]基于传统数字图像处理的清点方式在复杂环境中易受拍摄背景、拍摄角度、光照强度等多方面因素的干扰，出现重检、漏检等情况。随着人工智能的发展，深度学习技术在目标检测等众多领域中取得了重大突破，但钢筋作为分辨率低、体积小、数量多的物体，而很难被全部检测到。一般的目标检测技术密集检测性能差，主要是由于目标之间往往具有非常相似的特征，这就使得检测器无法有效区分两个检测框，另外非极大值抑制等后处理对重叠目标的检测结果也有较大影响。
[0004]因此，本专利技术提供了一种基于密集目标检测的钢筋计数处理方法，解决现有人力清点钢筋计数不准确的技术问题。

技术实现思路

[0005]本专利技术要解决的技术问题是：提供一种基于密集目标检测的钢筋计数处理方法，解决现有人力清点钢筋计数不准确的技术问题。
[0006]为实现上述目的，本专利技术采用的技术方案如下：一种基于密集目标检测的钢筋计数处理方法，包括以下步骤：步骤S1、采集钢筋切面图片，确定先验框；步骤S2、输入钢筋切面图片，构建RPN网络模型生成包含预测框的候选区域；步骤S3、构建分支网路，预测候选区域中预测框的类别得分和相对坐标；步骤S4、构建损失函数，基于带动量的随机梯度下降法更新RPN网络模型参数；步骤S5、将更新的RPN网络模型迭代反馈，得到钢筋检测计数模型；步骤S6、对钢筋检测计数模型输出的预测框进行软性非极大值抑制处理，过滤掉重复预测框，计算得到钢筋切面图片中钢筋的数量。
[0007]进一步地，RPN网络模型函数为：，其中为一组真实框，b
i
为RPN网络模型的预测框的分类和相对坐标值，IoU为预测框和真实框相交区域面积和相并区域面积的之比，θ为设定的IoU阈值，g
i
为第i个真实框的分类和相对坐标值。
[0008]进一步地，每一个候选区域构建一个分支网路。
[0009]进一步地，分支网路函数为：，其中k 为1到K之间的整数，c
k
为类标签的置信度，l
k
为分支网络预测的第i个预测框相对坐标值，K
为超参数、即候选区域保留最大的预测框数，b
i
为RPN网络模型的第i个预测框的分类和相对坐标值。
[0010]进一步地，损失函数为：，其中，α和β分别是分类误差和定位误差的调节系数，g
il
为第i个真实框的相对坐标值，g
ic
为第i个真实框的分类，x
ki
表示是否将第k个预测框分配给候选区域i，l
k
表示分支网络预测的预测框相对坐标值，c
k
代表类标签的置信度，BCE
loss
表示二分类交叉熵损失，b
i
为RPN网络模型的第i个预测框的分类和相对坐标值。
[0011]进一步地，钢筋切面图片中钢筋数量为10
‑
500。
[0012]进一步地，钢筋切面图片为钢筋整齐叠放状态下的端面图片。
[0013]进一步地，利用K
‑
means算法类聚生成先验框参数。
[0014]进一步地，候选区域的预测框为一个或者多个。
[0015]进一步地，采用F1分数计算钢筋检测计数模型中钢筋数量的召回率和准确率。
[0016]与现有技术相比，本专利技术具有以下有益效果：本专利技术设计科学合理，使用方便，解决了现有建筑工地中人力清点钢筋计数不准确的技术问题，为智慧工地建设发展助力。本专利技术提供了一种基于密集目标检测的钢筋计数处理方法，可由一种候选区域预测多个目标，有效适用于目标物体的密集检测；并基于动量的随机梯度下降法进行了模型更新，可优化与真实框匹配程度较高的预测框；再对更新模型输出的预测框，软性非极大值抑制处理，过滤掉重复预测框，最终得到图像中钢筋的数量。本专利技术提高了高检索率情况下的准确率，相较于现有其他主流模型，具有更好的平均精度、召回率和准确率。
附图说明
[0017]图1为本专利技术方法流程图。
具体实施方式
[0018]为了使本专利技术的目的、技术方案及优点更加清楚明白，以下结合附图，对本专利技术进一步详细说明。显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0019]如图1所示，本专利技术提供的一种基于密集目标检测的钢筋计数处理方法，解决现有人力清点钢筋计数不准确的技术问题，设计科学合理，使用方便，解决了现有建筑工地中人力清点钢筋计数不准确的技术问题，为智慧工地建设发展助力。本专利技术不仅可用于钢筋计数，还可应用于人群检测、车辆检测等密集场景。
[0020]本专利技术提供了一种基于密集目标检测的钢筋计数处理方法，可由一种候选区域预测多个目标，有效适用于目标物体的密集检测；并基于动量的随机梯度下降法进行了模型更新，可优化与真实框匹配程度较高的预测框；再对更新模型输出的预测框，软性非极大值
抑制处理，过滤掉重复预测框，最终得到图像中钢筋的数量。本专利技术提高了高检索率情况下的准确率，相较于现有其他主流模型，具有更好的平均精度、召回率和准确率。
[0021]本专利技术包括以下步骤：步骤S1、采集钢筋切面图片，确定先验框；步骤S2、输入钢筋切面图片，构建RPN网络模型生成包含预测框的候选区域；步骤S3、构建分支网路，预测候选区域中预测框的类别得分和相对坐标；步骤S4、构建损失函数，基于带动量的随机梯度下降法更新RPN网络模型参数；步骤S5、将更新的RPN网络模型迭代反馈，得到钢筋检测计数模型；步骤S6、对钢筋检测计数模型输出的预测框进行软性非极大值抑制处理，过滤掉重复预测框，计算得到钢筋切面图片中钢筋的数量。
[0022]在步骤S1的钢筋切面图片采集操作中，钢筋切面图片的钢筋数量为10
‑
500，钢筋切面图片为钢筋整齐叠放状态下的端面图片。
[0023]在步骤S1的先验框操作中，利用K
‑
means算法类聚生成先验框参数。先验框代表最有可能的目标物体的宽度和高度。
[0024]在步骤S2的RPN网络模型生成操作中，RPN网络模型函数为：，其中为一组真实框，b
i
为RPN网络模型的第i个预测框的分类和相对坐标值，IoU为预测框和真实框相交区域面积和相并区域面积的之比，θ为设定的IoU阈值，g
i
为第i个真实框的分类和相对坐标值。在传统的本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于密集目标检测的钢筋计数处理方法，其特征在于，包括以下步骤：步骤S1、采集钢筋切面图片，确定先验框；步骤S2、输入钢筋切面图片，构建RPN网络模型生成包含预测框的候选区域；步骤S3、构建分支网路，预测候选区域中预测框的类别得分和相对坐标；步骤S4、构建损失函数，基于带动量的随机梯度下降法更新RPN网络模型参数；步骤S5、将更新的RPN网络模型迭代反馈，得到钢筋检测计数模型；步骤S6、对钢筋检测计数模型输出的预测框进行软性非极大值抑制处理，过滤掉重复预测框，计算得到钢筋切面图片中钢筋的数量。2.根据权利要求1所述的一种基于密集目标检测的钢筋计数处理方法，其特征在于，RPN网络模型函数为：，其中为一组真实框，b
i
为RPN网络模型的第i个预测框的分类和相对坐标值，IoU为预测框和真实框相交区域面积和相并区域面积的之比，θ为设定的IoU阈值，g
i
为第i个真实框的分类和相对坐标值。3.根据权利要求1所述的一种基于密集目标检测的钢筋计数处理方法，其特征在于，每一个候选区域构建一个分支网路。4.根据权利要求1所述的一种基于密集目标检测的钢筋计数处理方法，其特征在于，分支网路函数为：，其中k 为1到K之间的整数，c
k
为类标签的置信度，l
k
为分支网络预测的预测框相对坐标值，K为超参数、即候选区域保留最大的预测框数，b
i
为RPN网络模型的第i...

【专利技术属性】
技术研发人员：谷满昌，张镜，
申请(专利权)人：中建电子商务有限责任公司，
类型：发明
国别省市：