[发明专利]一种基于模型剪枝的桥梁病害实时检测方法

摘要

本发明公开了一种基于模型剪枝的桥梁病害实时检测方法，首先利用无人机获取桥梁表观害图片，然后按照比例将数据集划分为训练集、验证集和测试集；接着运用图像增强方法，增加数据集的数量；构建yolov3网络框架，对网络的每一个批归一化层(BN)添加γ参数项的L1正则化约束，设置渐进式局部衰减的稀疏度调整策略，对改造后的网络进行训练，得到训练模型；得到训练好的所有BN层的γ参数之后，根据层间和层内的γ参数大小进行层和通道的模型剪枝，删除权重较小的通道，得到剪枝后的模型；利用训练好的模型进行桥梁表观病害自动识别。本发明效率高，成本低，相对于传统的人工标记标签训练法，更具有明显的自动化和高效性。