基于改进的Swin Transformer的输油管道微小缺陷检测模型

针对磁漏微小缺陷检测困难、检测数据不足、检测精度低等问题,提出了去噪扩散概率模型(DDPM)-门控并行膨胀卷积的移位窗口变换器(DGPST)。文中引入了DDPM作为数据生成模型,获得高质量的数据样本。门控并行膨胀卷积的移位窗口变换器(GPST)是在移位窗口变换器(Swin-Transformer)的基础上进行改进的。在GPST的主干网络中引入了门控并行膨胀卷积层,门控单元可以减少不必要的计算量,卷积提高了网络提取全局上下文信息的能力,有利于提高模型检测精度。实验结果表明:当IOU=0.5时,该算法的精确度为95.6%,推理时间(Latency)为8.2 ms。