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

一种轻量化的多目标实时检测模型
引用本文:邱博,刘翔,石蕴玉,尚岩峰.一种轻量化的多目标实时检测模型[J].北京航空航天大学学报,2020,46(9):1778-1785.
作者姓名:邱博  刘翔  石蕴玉  尚岩峰
作者单位:1.上海工程技术大学 电子电气工程学院, 上海 201620
基金项目:国家重点研发计划2016YFC0801304上海市“科技创新行动计划”高新技术领域项目17511106803
摘    要:为实现公安监控系统内容分析的精准智能及提高服务实战能力,提出一种轻量化的多目标实时检测算法。首先,基于CenterNet检测网络增加了CBNet的多融合阶梯级联结构,有效地解决了主干网络在日常监控中特征提取能力不足的问题;其次,通过模型剪枝压缩网络减少参数量,加快了监控视频分析速度。本文利用部分COCO数据集和自行采集的现场数据进行训练与测试,并与其他主流检测算法(YOLO、Faster-RCNN、SSD等)进行消融实验。实验结果表明:所提模型在公共安全监控中能有效地做到速度与精度的均衡,并具有较强的普适性。 

关 键 词:目标检测    深度学习    模型压缩    模型蒸馏    级联融合
收稿时间:2020-03-02

A lightweight multi-target real-time detection model
Institution:1.School of Electronic and Electrical Engineering, Shanghai University of Engineering Science, Shanghai 201620, China2.Internet of Things Technology R & D Center, The Third Research Institute of the Ministry of Public Security, Shanghai 200031, China
Abstract:For the public security monitoring system, a lightweight multi-target real-time detection algorithm is proposed in order to realize the accurate intelligence of the content analysis and improve the actual service ability. First, the multi-fusion gradient cascade structure of CBNet is added based on CenterNet detection network, which effectively solves the problem of insufficient feature extraction capability of the backbone network in daily monitoring videos. Second, the number of parameters is reduced through the model pruning and compression, which can speed up the analysis speed of monitoring videos. During the experiments, the dataset for training and testing consists of a part of COCO datasets and a number of field data collected by ourselves. The ablation experiments are conducted with other mainstream detection algorithms (YOLO, Faster-RCNN, SSD, etc.). The experimental results show that the presented model can effectively balance the speed and precision in the analysis of monitoring videos for public security and has stronger universality. 
Keywords:
点击此处可从《北京航空航天大学学报》浏览原始摘要信息
点击此处可从《北京航空航天大学学报》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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