混叠场景下的轻量级口罩佩戴检测算法 |
| |
作者姓名: | 安祯阳 |
| |
作者单位: | 重庆交通大学机电与车辆工程学院 |
| |
摘 要: | 针对混叠场景下口罩佩戴检测识别率低,而现有检测模型结构复杂难以部署的难题,提出了一种轻量级口罩佩戴检测算法。首先,轻量化网络MobileNetv3作为混叠场景图像的特征提取网络;其次提出通道混洗,空间上利用不同感受野的卷积核进行特征提取的注意力机制,实现特征信息的强化;最后设计了损失函数解决了数据类不平衡问题,提高了模型检测精度。在公开数据集测试表明,模型平均检测精度为78.1%,FPS达到65.53 Hz,满足在小型设备部署的要求。
|
关 键 词: | 轻量级 目标检测 部分卷积 特征融合 |
|
|