一种基于自注意力机制的多类管道缺陷检测、追踪与计数方法
摘要:
本发明公开了一种基于自注意力机制的多类管道缺陷检测、追踪与计数方法,包括如下步骤:S1、获取多类管道缺陷图像;S2、建立管道缺陷图像数据集;S3、构建管道缺陷检测模型并训练;S4、构建管道缺陷追踪与计数模型;S5、测试并输出结果。本发明采用基于自注意力机制的深度学习方法,研发适用于管道缺陷特征的目标检测、追踪与计数算法,基于管道缺陷大数据进行模型训练,提高模型的鲁棒性和泛化能力。本发明采用Transformer基础网络代替了传统卷积神经网络,提高模型检测的准确度,实现了管道缺陷的自动识别与计数功能。
0/0