基于改进注意力机制和NetVLAD的弹壳痕迹特征识别

Feature Recognition of Cartridge Case Traces Based on Improved Attention Mechanism and NetVLAD

  • 摘要: 针对传统弹壳痕迹识别方法大多停留在人工识别阶段,使得弹壳痕迹识别准确率存在极大的缺陷的问题,提出了一种基于改进NetVLAD网络的弹壳痕迹特征识别方法。首先,采集了7类枪种的弹壳痕迹图像并建立图像数据库,以NetVLAD网络为基础模型,ResNet-18为主干网络;然后,引入了改进注意力机制CBAM(Convolutional Block Attention Module);最后,针对卷积核尺度单一问题,加入1维卷积和多尺度空洞卷积模块对其进行改进,融合了不同感受野的特征,在降维的同时有效抑制特征损耗,最终网络模型生成更加准确且紧凑的VLAD(Vector of Locally Aggregated Descriptors)特征描述子用于痕迹图像间的特征匹配,从而提高图像检索准确率。实验结果表明,所提模型在7类枪种的弹壳痕迹数据集上有较好的识别和匹配能力,准确率在95%以上;与其他神经网络对比,本文模型对比VGG-16、ResNet-50、ResNet-18、AlexNet、InceptionV3作为主干网络的NetVLAD模型准确率分别提高19.07%、17.16%、14.58%、16.51%和21.47%。所提模型对比改进前的模型在7类枪种的弹壳痕迹数据集上的匹配准确率均有所提高,可为弹壳痕迹鉴定领域枪种分类提供参考。

     

    Abstract: Aiming at the traditional cartridge case trace recognition methods mostly stay in the manual recognition stage, which makes the accuracy of cartridge case trace recognition has great defects, we propose a cartridge case trace feature recognition method based on the improved NetVLAD network. Firstly, we collect images of cartridge case traces of seven types of establish guns and an image database, with NetVLAD network as the base model and ResNet-18 as the backbone network. Secondly, we introduce the improved attention mechanism CBAM (Convolutional block attention module). Finally, for the convolutional kernel scale single problem, we add one-dimensional convolution and multi-scale cavity convolution module to improve it, fuse different sensory field features, in order to reduce the dimensionality and also effectively inhibit the feature loss. The final network model generates more accurate and compact VLAD (Vector of Locally Aggregated Descriptors), which are used for feature matching between trace images, thus improving the image retrieval accuracy. The experimental results show that the proposed model has a better recognition and matching ability on the cartridge case trace dataset of 7 types of guns, with an accuracy of more than 95%, and compared with other neural networks, the proposed model compares the accuracy of the NetVLAD model with VGG-16, ResNet-50, ResNet-18, AlexNet, and InceptionV3 as the backbone network by 19.07%, 17.07%, 17.07%, and 17.07%, respectively, improved by 19.07%, 17.16%, 14.58%, 16.51% and 21.47%. Moreover, the matching accuracy of the proposed model is improved compared with the pre-improved model on the cartridge traces dataset of seven types of guns, which can provide a reference for the classification of gun types in the field of cartridge traces identification.

     

/

返回文章
返回