首页 | 本学科首页   官方微博 | 高级检索  
     

基于多任务级联残差网络的枪支图像识别系统
引用本文:周志飞,吴金龙,李轶昳,贾力榜,沈玉杰,张刚,崔斌.基于多任务级联残差网络的枪支图像识别系统[J].计算机工程,2022,48(1):214-219.
作者姓名:周志飞  吴金龙  李轶昳  贾力榜  沈玉杰  张刚  崔斌
作者单位:1. 公安部物证鉴定中心, 北京 100038;2. 北京多维视通技术有限公司, 北京 100070;3. 中国科学院自动化研究所, 北京 100190
基金项目:中央级公益性科研院所基本科研业务费专项资金(2018JB020);
摘    要:针对枪支种属识别目前依赖检验人员经验、识别效率较低的问题,建立一种基于多任务级联深度残差网络的枪支图像自动识别模型。以ResNet18为基本构建单元,通过级联融合4个任务中的Softmax损失函数约束,实现对枪支图像从枪族到枪型的多维度聚类。在该模型的基础上,设计一套制式枪支图像智能检索系统,对拍摄上传的枪支图像种属信息进行自动识别。在自建的制式枪支图像数据集上进行实验,结果表明,与EfficientNet、NTS-net等模型相比,该模型的识别准确率更高,Rank-1、Rank-20识别准确率分别达到61.12%、95.28%,且其具有更好的鲁棒性。

关 键 词:枪支种属识别  深度学习  残差网络  细粒度图像识别  数据增广  
收稿时间:2020-10-23
修稿时间:2021-01-13

Firearm Image Recognition System Based on Multi-Task Cascaded Residual Network
ZHOU Zhifei,WU Jinlong,LI Yiyi,JIA Libang,SHEN Yujie,ZHANG Gang,CUI Bin.Firearm Image Recognition System Based on Multi-Task Cascaded Residual Network[J].Computer Engineering,2022,48(1):214-219.
Authors:ZHOU Zhifei  WU Jinlong  LI Yiyi  JIA Libang  SHEN Yujie  ZHANG Gang  CUI Bin
Affiliation:1. Institute of Forensic Science of China, Beijing 100038, China;2. Beijing Visystem Co., Ltd., Beijing 100070, China;3. Institute of Automation, Chinese Academy of Sciences, Beijing 100190, China
Abstract:The traditional firearm recognition methods rely heavily on expertise, and are limited in recognition accuracy.To address the problem, a model for automatic firearm image recognition is built using multi-task cascaded deep residual network.With ResNet18 as the basic build block, this model fuses the Softmax loss function constraints in the four cascaded tasks and realizes firearm image clustering, which is based on multiple dimensions ranging from the firearm family to the specific gun type.Based on the proposed model, a system for intelligent firearm image retrieval is designed, which can automatically recognize the type of the firearms in uploaded images.The experimental results on a self-made firearm image dataset show that the model displays a higher recognition accuracy in Rank-1(61.12%) and Rank-20(95.28%) than EfficientNet, NTS-net and other models.The proposed model also provides better robustness for gun image recognition in real scenes.
Keywords:firearm species recognition  deep learning  residual network  fine-grained image recognition  data augmentation
本文献已被 维普 万方数据 等数据库收录!
点击此处可从《计算机工程》浏览原始摘要信息
点击此处可从《计算机工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号