王伟志, 胡海洋, 顾盼, 崔高宾, 李正华, 李渊
铆钉在工业领域中常被用于金属零件或者结构体的连接,一旦出现漏装,就可能发生严重的安全事故。为了解决上述的问题,提出了一种用于铆钉连接检测的新模型LRD-DETR。模型在提取网络进行特征提取后,加入了一个多尺度融合模块DSPANet,加强了特征图的语义信息;在编码器中设计了一种限制范围的可变形注意力机制,限制了每一个参考点选取采样点的范围;在解码器中使用了DAB-DETR的解码器部分,并在其基础上设计了一种新的MLP(Multilayer Perceptron)模块用于更新每一次的4D锚框信息。在自建的铆钉数据集中,模型取得了令人满意的结果,mAP值达到了59.2%,实验结果表明,相比于主流的目标检测算法,本文提出的铆钉检测方法效果更优。