[发明专利]基于深度神经网络的钢筋数量检测方法及系统

权利要求书

1.一种基于深度神经网络的钢筋数量检测方法，其特征在于，包括以下步骤：

S100，根据第三方物体检测数据集，基于以ResNeXt、FPN、RPN和RoiAlign为主干网络的Faster-RCNN算法，生成预训练模型；

S200，输入包括标注信息的钢筋数据集，对所述钢筋数据集中的图片进行第一数据增强转换，对所述预训练模型进行迁移学习训练，生成钢筋检测模型；

S300，获取包括钢筋横截面的待检测图片，通过所述钢筋检测模型得到第一检测框，以及，对所述待检测图片进行第二数据增强转换，得到若干第二增强图片，通过所述钢筋检测模型得到第二检测框；

S400，根据坐标信息匹配所述第一检测框及所述第二检测框，基于钢筋置信度及检测框重叠度，得到钢筋检测数量。

2.根据权利要求1所述的基于深度神经网络的钢筋数量检测方法，其特征在于，所述步骤S200包括：

S210，将所述预训练模型中的分类输出层神经元数修改为两个；

S220，加载所述钢筋数据集，基于预设概率对所述钢筋数据集中的图片进行所述第一数据增强转换，得到第一增强图片；

S230，根据所述第一增强图片及所述标注信息，对所述预训练模型进行学习训练，得到所述钢筋检测模型。

3.根据权利要求1所述的基于深度神经网络的钢筋数量检测方法，其特征在于，所述第一数据增强转换的方法包括：根据预设规则选择中心裁剪、水平翻转、按短边缩放、随机高斯模糊及亮度调整的其中一项处理图片。

4.根据权利要求1所述的基于深度神经网络的钢筋数量检测方法，其特征在于，所述第二数据增强转换的方法包括：选择水平翻转、按短边缩放图片及低照度图像增强中的其中至少一项处理图片。

5.根据权利要求1所述的基于深度神经网络的钢筋数量检测方法，其特征在于，所述步骤S400包括：

S410，通过所述第二增强图片与所述待检测图片的转换映射关系，得到所述第二检测框的所述坐标信息；

S420，根据所述坐标信息，从所述第二检测框中获取所述第一检测框的候选匹配检测框；

S430，根据钢筋置信度，从所述第一检测框及所述候选匹配检测框中选出基准框及非基准框；

S440，基于交并比计算所述基准框及所述非基准框的重叠度，根据重叠度舍弃所述非基准框。

6.根据权利要求5所述的基于深度神经网络的钢筋数量检测方法，其特征在于，所述步骤S430包括：

S431，从所述第一检测框及所述候选匹配检测框中，筛选出所述钢筋置信度大于第一阈值的检测框，得到若干第三检测框；

S432，从所述第三检测框中选取所述钢筋置信度最高的检测框为所述基准框，剩余的所述第三检测框则为非基准框。

7.根据权利要求5所述的基于深度神经网络的钢筋数量检测方法，其特征在于，所述步骤S440包括：基于交并比计算所述基准框与所述非基准框的重叠度，若所述重叠度大于第二阈值，则舍弃所述非基准框；否则，保留所述基准框及所述非基准框。

8.一种基于深度神经网络的钢筋数量检测系统，使用权利要求1至7中任一项的方法，其特征在于，包括：

预训练模型生成模块，用于根据第三方物体检测数据集，基于以ResNeXt、FPN、RPN和RoiAlign为主干网络的Faster-RCNN算法，生成预训练模型；

检测模型生成模块，用于输入包括标注信息的钢筋数据集，对所述钢筋数据集中的图片进行第一数据增强转换，对所述预训练模型进行迁移学习训练，生成钢筋检测模型；

检测处理模块，用于获取包括钢筋横截面的待检测图片，通过所述钢筋检测模型得到第一检测框，以及，对所述待检测图片进行第二数据增强转换，得到若干第二增强图片，通过所述钢筋检测模型得到第二检测框；

数量计算模块，用于根据坐标信息匹配所述第一检测框及所述第二检测框，基于钢筋置信度及检测框重叠度，得到钢筋检测数量。