中国科学技术大学学报 ›› 2020, Vol. 50 ›› Issue (1): 79-85.DOI: 10.3969/j.issn.0253-2778.2020.01.010

• 科研论文 • 上一篇    

基于自学习的行人检测方法

汪 中   

  1. 1.合肥师范学院计算机学院,安徽合肥 230601;2.大数据分析与应用安徽省重点实验室,中国科学技术大学,安徽合肥 230027
  • 收稿日期:2019-11-03 修回日期:2020-01-10 出版日期:2020-01-31 发布日期:2020-01-31
  • 通讯作者: 刘贵全
  • 作者简介:汪中,男,1984年生, 博士/高级工程师.研究方向:行人检测、人工智能. E-mail: zhongw@ustc.edu.cn
  • 基金资助:
    国家自然科学基金(61976198),安徽高校自然科学研究重点项目(KJ2018A0498,KJ2019A0726),大数据分析与应用安徽省重点实验室开放课题资助.

A self-learning framework for pedestrian detection

WANG Zhong   

  1. 1. School of Computer Science and Technology, Hefei Normal University, Hefei 230601, China; 2. Anhui Province Key Laboratory of Big Data Analysis and Application, University of Science and Technology of China, Hefei 230027,China
  • Received:2019-11-03 Revised:2020-01-10 Online:2020-01-31 Published:2020-01-31

摘要: 离线训练的分类器应用于特定场景时,其检测性能将急剧下降.手工标注虽然可以提高检测性能,但是需要耗费大量的人工成本.为此提出一种基于自学习的行人检测方法,该方法可以改变任意离线训练的分类器用于特定场景的行人检测,并且取得了较好的识别率.首先将训练级联分类器作为离线分类器,并使用任意公开的行人图片训练高斯混合模型(GMM);然后利用离线分类测器对特定场景进行行人检测并获取候选对象的置信分数;再根据置信分数的高低构建正负样本集合并使用高斯混合模型重新标识样本;最后使用SVM分类器在线训练行人分类器,对候选对象进行重新预测.在公开和自制数据集上的实验结果表明,该方法提高了通用行人检测器的准确性,并且明显优于传统方法.

关键词: 行人检测, 自学习, 分类器, 高斯混合模型, 梯度直方图

Abstract: The performance of offline trained pedestrian detectors significantly drops when they are applied to the specific scene. Although manual labeling can improve detection performance, it requires a lot of human effort. In this paper, a self-learning framework is proposed for pedestrian detection, which can adapt any offline trained detector to a specific scene and obtain a better performance. Firstly, Cascade classifier is used as an offline classifier, while a Gaussian Mixture Model (GMM) is trained using a set of public pedestrian photos. Next, a low threshold offline classifier is used to perform pedestrian detection on a specific scene and the confidence score of candidate detections is obtained. Then, samples with high confidence scores are selected as positive samples, while those with low confidence scores are taken as a negative samples, and GMM is used to represent the candidate detection again. Finally, a discriminative pedestrian classifier is trained online using the SVM classifier to re-estimate candidate objects. Experimental results on public and self-made datasets show that the proposed method can improve the accuracy of the generic pedestrian detector and significantly outperforms the traditional methods.

Key words: pedestrian detection, self-learning, classifier, Gaussian mixture model, histogram of gradients