一种基于RT-DETR改进的板对板连接器表面缺陷检测方法

本发明提出了一种基于RT‑DETR改进的板对板连接器表面缺陷检测方法。该方法包括采集板对板连接器表面缺陷图像，对图像进行预处理，构建缺陷数据集；针对RT‑DETR模型进行改进，在主干网络中，设计了FBConv模块和MlpBlock模块分别替换首层卷积层和第四个BasicBlock模块，在高效混合编码器中，使用RepNCSPELAN4特征提取模块替换原有的三个RepC3模块，并设计轻量化RepFB模块替换最后一个RepC3模块，引入特征金字塔网络BiFPN替换Concat模块，使用SCDown模块替换部分卷积层进行空间下采样；最后，将原GIoU损失函数替换为Inner‑MPDIoU损失函数，并将改进后的模型命名为BTB‑RTDETR；将BTB‑RTDETR模型进行训练，用于板对板连接器表面缺陷的检测。本发明有效提高了对板对板连接器工件表面缺陷的检测准确性和检测效率，降低了人力成本。