[发明专利]一种基于U-net网络和SC-SAM注意力机制的大坝裂缝检测方法

说明书

本发明公开了一种基于U‑net网络和SC‑SAM注意力机制的大坝裂缝检测方法，该方法首先收集大坝数据集，并对该数据集进行数据扩充。随后构建深度学习分割网络U‑net模型，并在此基础之上添加SC‑SAM注意力机制，该注意力机制由两个部分构成，CAM提高了特征图中的裂缝通道权重，SAM提高了特征图中空间域上裂缝区域的权重，两者的相互协作使得模型对于大坝裂缝检测的准确度有了极大的提高。将扩充后的数据集分为训练集和测试集，利用训练集带SC‑SAM注意力机制的深度学习分割网络模型进行训练，得到训练好的模型；根据训练好的模型，将测试集输入训练好的模型中进行测试，得到大坝裂缝测试结果。

技术领域

本发明涉及一种基于U-net网络和SC-SAM注意力机制的大坝裂缝检测方法，属于图像识别技术领域。

背景技术

大坝是我国水利建设中的重要组成部分，大坝的安全稳定对于水利建设的发展有着关键性的作用，因此如何高效准确的检测出大坝存在的问题是我们需要面临的一个挑战。基于大坝的基本性质，由于大坝长时间受到水压和雨水冲刷、渗透、侵蚀的影响，极易产生裂缝，而裂缝的存在会导致大坝建筑的恶化、破损与崩塌，因此在大坝存在的各种问题中，裂缝的检测是最需要解决的。

对于裂缝的检测方法有很多，如赵芳等人在改进的Canny算子在裂缝检测中的应用一文中针对于噪声与边缘检测精度问题，提出了一种新的Canny边缘检测方法(将多尺度形态学和双边滤波结合)；周慧媛等人在基于对比度受限自适应直方图多种路面裂缝检测与识别一文中提出了一种新的路面检测与识别方法(利用CLAHE和中值滤波去噪，利用形态学去伪裂缝)；韦春桃等人基于自适应阈值的细小裂缝与微灰度差异裂缝自动检测方法在一文中针对路面各种噪声以及细小裂缝和微灰度差异问题，提出了一种基于自适应阈值的裂缝自动检测方法；李良福等人利用滑动窗口算法将桥梁裂缝图像切分为较小的桥梁裂缝面元图像和桥梁背景面元图像,并根据对面元图像的分析,提出一种基于卷积神经网络的DBCC(Deep bridge crack classify)分类模型；Shenyuan Li等人针对混凝土表面包含各种类型噪声带来的问题，提出了一种使用深度卷积神经网络(CNN)的基于图像的裂纹检测方法。

虽然这些检测方法并没有应用到大坝裂缝的检测中，但是这些方法中提出的一些观点、想法可以在我们检测大坝裂缝图像时提供一些帮助。如Shuang Wang等人利用整幅图像重建三维点云模型，然后利用FCN对标注数据进行迭代训练，得到裂纹检测模型，遍历剩余的原始尺寸图像进行裂纹预测；Jianghong Tang等人为了提高Faster R-CNN模型对多个小目标的检测精度，提出了一种基于Faster R-CNN(ME-Faster R-CNN)的多任务增强大坝裂缝图像检测方法，以适应不同光照环境和长度的大坝裂缝检测；徐辉等人针对传统的裂缝检测算法检测大坝裂缝不能很好地克服水下噪声的问题,提出一种基于图像显著性检测大坝裂缝的方法；蒋小燕等人面对大坝裂缝图像模糊不清、亮度不均、对比度低、随机噪声大等特点，提出了一种基于格子波尔兹曼模型(LBM)大坝裂缝检测方法；由于大坝缺陷图像具有信噪比低、光照分布极度不均匀等特征,分类识别算法的识别率较低，因此毛莺池等人提出了一种基于图像LBP特征和Gabor特征组合与CNN相结合的缺陷图像识别方法。

发明内容

本发明所要解决的技术问题是：提供一种基于U-net网络和SC-SAM注意力机制的大坝裂缝检测方法，在原始U-net网络的基础上，添加SC-SAM注意力机制，分别从通道域和空间域两个方面对裂缝所在区域进行了权重的提升，为模型识别分割大坝裂缝提供了一种有效的方案。

本发明为解决上述技术问题采用以下技术方案：

一种基于U-net网络和SC-SAM注意力机制的大坝裂缝检测方法，包括如下步骤：

步骤1，获取大坝裂缝数据集，包括大坝裂缝的真实图片和与之对应的标签图片，利用图像数据增强技术对大坝数据集进行扩充，得到大坝裂缝样本集；

步骤2，构建SC-SAM注意力机制的深度学习分割网络模型；具体如下：