摘要
本发明提供一种改进DETR的管道缺陷检测方法,涉及计算机视觉目标检测技术领域,包括:首先,DETR的主干网络修改为CrossFormer;其次,编码器通过多尺度可变形注意力对主干网络输出的多尺度特征进行增强;之后,解码器将编码器输出的特征与一组预定义的目标查询相结合,生成检测头进行最终结果预测的输入;然后,检测头预测目标的类别和边界框坐标;最后,使用warmup和cosine的学习率调整策略,并且使用Adamw优化器训练改进的DETR网络。本发明的基于DETR的改进方法能够有效提高管道缺陷检测的平均准确率。