|
|||||
|
|
| 基于多层注意力机制的口罩佩戴检测算法改进 | |
| 引用本文： | 周蕾,陈冠宇,钟海莲.基于多层注意力机制的口罩佩戴检测算法改进[J].电子器件,2023(6):1652-1660. |
| 作者姓名： | 周蕾 陈冠宇 钟海莲 |
| 作者单位： | 1. 淮阴工学院计算机与软件工程学院;2. 淮阴工学院化学工程学院 |
| 基金项目： | 江苏省教育厅自然科学基金项目（20KJA520008）； |
| 摘 要： | 针对已有的对于口罩的目标检测算法中存在的问题，如对小目标检测精度低，复杂场景下出现漏检等问题，提出了一种基于多层注意力机制的口罩佩戴检测算法YOLOX-l-sd。通过在骨干网络中添加Swin Transformer的特征层来提升目标检测算法的性能，加入多层注意力机制提高特征提取效果以及减少计算量，采用DIOU损失函数提升模型的精度，采用DW卷积以进一步减少计算量。实验结果表明，相比原YOLOX-l模型，改进后的模型在精度上提高了2.45%，达到了预期的效果。 |
| 关 键 词： | YOLOX Swin Transformer 口罩检测 多层注意力机制 |