一种架空管道缺陷识别的方法、装置、设备及介质

本申请提供了一种架空管道缺陷识别的方法、装置、设备及介质，该方法包括：模型训练阶段：对数据集中的原始缺陷图像进行数据增强得到多组增强缺陷图像；其中，增强缺陷图像包括第一部分图像和第二部分图像；使用第一部分图像的第一特征向量构建分类分支子模型的第一损失函数；使用第二部分图像的第二特征向量构建表示学习分支子模型的第二损失函数；其中，第一损失函数和第二损失函数用于构建整体损失函数；通过对整体损失函数进行参数调整，得到缺陷识别模型；模型使用阶段：将待测架空管道的待测图像输入到缺陷识别模型中，得到缺陷识别模型输出的缺陷识别结果。本申请显著了提升了长尾分布下管道缺陷的识别能力、识别精度。