基于级联自注意力的轻量级实时PCB缺陷检测算法研究

摘要

LHsRT-DETR(lightweight high-speed real-time detection transformer)是一种PCB缺陷检测轻量级模型，为提升模型的实时性，减少其参数量，对模型进行改进。首先，提出HsCAttention机制，通过对自注意力机制的修改来替代传统模块；其次，引入ContextGuided网络以增强特征提取能力和设计主干的轻量化；最后，加入剪枝编码器结构，减少冗余并降低通道维度和参数个数。实验结果表明，相比基准模型，LHsRT-DETR在PKU-PCB缺陷数据集上达到96.5%的检测精度，同时将模型参数降低68.65%,计算量从58.6 G降至21.8 G,达到424.8 FPS的推理速度。通过对模型大小、计算效率、推理速度和检测精度的综合评估，LHsRT-DETR在当前众多模型中具有显著优势。

关键词

PCB / 目标检测 / transformer / RT-DETR / 自注意力机制