基于注意力机制的改进YOLOX行人跌倒检测方法 |
| |
引用本文: | 周蕾,钟海莲,陈冠宇.基于注意力机制的改进YOLOX行人跌倒检测方法[J].电子器件,2023(2):404-413. |
| |
作者姓名: | 周蕾 钟海莲 陈冠宇 |
| |
作者单位: | 1. 淮阴工学院计算机与软件工程学院;2. 淮阴工学院化学工程学院 |
| |
基金项目: | 江苏省教育厅自然科学基金项目(20KJA520008); |
| |
摘 要: | 针对现有行人跌倒检测算法在复杂场景下存在漏检、检测精度低等问题,提出一种基于注意力机制的行人跌倒检测方法 YOLOX-s-EsE。改进的模型在骨干网络中引入SimAM注意力模块,在Bottleneck和特征融合模块增加ECA通道注意力模块,以进一步提取特征层的关键信息,损失函数采用EIo U,可以更有效地计算出预测框和真实框的差距,提升模型的精度。实验结果表明,改进后的算法在复杂环境下目标的检测效果有了明显的提升,相比原YOLOX-s模型,算法的mAP提高了约1.8%,达到了89.23%,精度提高了约4.6%,达到了91.79%。
|
关 键 词: | YOLOX 注意力机制 行人检测 跌倒检测 损失函数 |
|