激光与光电子学进展, 2022, 59 (2): 0215004, 网络出版: 2021-12-29   

基于注意力机制的鞋型识别算法

Shoe Type Recognition Algorithm Based on Attention Mechanism
作者单位
中国人民公安大学侦查学院,北京 100038
摘要
根据现场遗留鞋印推断出作案人所穿鞋型,再从周围监控视频中搜索嫌疑鞋型已成为公安机关侦破案件的重要技战法。针对该技战法完全依赖人工筛查、受主观影响大、易造成漏检等问题,提出了一种基于注意力机制的鞋型识别算法。首先,建立了贴近公安刑侦实战、样本容量为300的多背景监控鞋型数据集。然后,提出了一种注意力机制模型,以增强残差网络(ResNet50)对鞋子重要特征的提取能力。最后,对比了选取不同特征层输出作为鞋子特征及不同卷积特征聚合方法对识别精度的影响。为了增强模型的泛化能力,在损失函数中加入Label Smoothing。在多背景数据集上的实验结果表明,本算法的Rank-1、平均精度均值分别达到74.32%和56.97%。
Abstract
It has become an important technique and tactics for the public security organs to infer the type of shoes worn by the perpetrators according to the shoe prints left at the scene, and then search the suspected type of shoes in the surrounding surveillance video. This technique is completely dependent on manual screening, which is greatly affected by subjective factors and easily leads to problems such as missed detection. To solve this problem, this paper proposes a shoe type recognition algorithm based on attention mechanism. First, close to the actual combat of public security criminal investigation, a multi background monitoring shoe data set with sample size of 300 is established. Then, an attention mechanism model is proposed to enhance the ability of the residual network (ResNet50) to extract important features of shoes. Finally, the effects of selecting the output of different feature layers as shoe features and different convolution feature aggregation methods on the recognition accuracy are compared. In order to enhance the generalization ability of the model, label smoothing is added to the loss function. The experimental results on the multi background data set show that the Rank-1 and mean average precision of the algorithm are 74.32% and 56.97%, respectively.

张家钧, 唐云祁, 杨智雄, 耿鹏志. 基于注意力机制的鞋型识别算法[J]. 激光与光电子学进展, 2022, 59(2): 0215004. Jiajun Zhang, Yunqi Tang, Zhixiong Yang, Pengzhi Geng. Shoe Type Recognition Algorithm Based on Attention Mechanism[J]. Laser & Optoelectronics Progress, 2022, 59(2): 0215004.

本文已被 1 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!