一种基于注意力机制的抗遮挡行人检测方法

摘要

本发明公开一种基于注意力机制的抗遮挡行人检测方法，步骤是：得到行人图片的多层卷积特征，以新型特征金字塔的结构融合，生成包含语义信息和细节信息的新特征层；采用k均值聚类得到预设的行人边界框的尺寸；预测所有行人边界框的回归偏置和分类置信度；计算交叉熵损失函数和改进的自适应遮挡感知回归损失函数，进而得到整体损失函数；迭代优化，不断训练得到检测模型；将待检测图片送入检测模型中，采用非极大值抑制法去除重叠程度高的框，最终得到检测结果。此种方法可改善对遮挡行人的检测能力，具有更高的检测精度和鲁棒性。