计算机应用 ›› 2017, Vol. 37 ›› Issue (11): 3176-3181.DOI: 10.11772/j.issn.1001-9081.2017.11.3176

• 第十六届中国机器学习会议(CCML 2017) • 上一篇    下一篇

基于深度迁移学习的烟雾识别方法

王文朋1, 毛文涛1,2, 何建樑1, 窦智1,2   

  1. 1. 河南师范大学 计算机与信息工程学院, 河南 新乡 453007;
    2. 河南省高校计算智能与数据挖掘工程技术中心, 河南 新乡 453007
  • 收稿日期:2017-05-16 修回日期:2017-06-07 出版日期:2017-11-10 发布日期:2017-11-11
  • 通讯作者: 毛文涛
  • 作者简介:王文朋(1989-),男,河南新乡人,硕士研究生,主要研究方向:机器学习、模式识别;毛文涛(1980-),男,河南新乡人,副教授,博士,CCF会员,主要研究领域为机器学习、弱信号检测;何建樑(1993-),男,河南信阳人,硕士研究生,主要研究方向:机器学习、时间序列预测;窦智(1977-),男,河南新乡人,副教授,博士,主要研究方向:图像处理、模式识别。
  • 基金资助:
    国家自然科学基金资助项目(U1204609);河南省高校科技创新人才支持计划(15HASTIT022);河南省高校青年骨干教师资助计划(2014GGJS-046);河南师范大学优秀青年科学基金资助项目(14YQ007);河南省高等学校重点科研项目计划(15A520078);河南省科技攻关项目(172102210333)。

Smoke recognition based on deep transfer learning

WANG Wenpeng1, MAO Wentao1,2, HE Jianliang1, DOU Zhi1,2   

  1. 1. College of Computer and Information Engineering, Henan Normal University, Xinxiang Henan 453007, China;
    2. Engineering Technology Research Center for Computing Intelligence and Data Mining of Henan Province, Xinxiang Henan 453007, China
  • Received:2017-05-16 Revised:2017-06-07 Online:2017-11-10 Published:2017-11-11
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (U1204609), the Program for Science & Technology Innovation Talents in University of Henan Province (15HASTIT022), the Funding Scheme of University Young Core Instructors in Henan Province (2014GGJS-046), the Foundation of Henan Normal University for Excellent Young Teachers (14YQ007), the Key Scientific Research Project in University of Henan Province (15A520078), the Key Scientific and Technological Project of Henan Province (172102210333).

摘要: 针对传统的基于传感器和图像特征的烟雾识别方法易被外部环境干扰且识别场景单一,从而造成烟雾识别精度较低,而基于深度学习的识别方法对数据量要求较高,对于烟雾数据缺失或数据来源受限的情况模型识别能力较弱的问题,提出一种基于深度迁移学习的烟雾识别方法。将ImageNet数据集作为源数据,利用VGG-16模型进行基于同构数据下的特征迁移。首先,将所有的图像数据进行预处理,对每张图像作随机变换(随机旋转、剪切、翻转等);其次,引入VGG-16网络,将其卷积层特征进行迁移,并连接预先使用烟雾数据在VGG-16网络中训练过的全连接层;进而构建出基于迁移学习的深度网络,从而训练得到烟雾识别模型。利用公开数据集以及真实场景烟雾图像进行实验验证,实验结果表明,和现有主流烟雾图像识别方法相比,所提方法有较高的烟雾识别率,实验精度达96%以上。

关键词: 深度学习, 迁移学习, 烟雾识别, 微量数据集

Abstract: For smoke recognition problem, the traditional recognition methods based on sensor and image feature are easily affected by the external environment, which would lead to low recognition precision if the flame scene and type change. The recognition method based on deep learning requires a large amount of data, so the model recognition ability is weak when the smoke data is missing or the data source is restricted. To overcome these drawbacks, a new smoke recognition method based on deep transfer learning was proposed. The main idea was to conduct smoke feature transfer by means of VGG-16 (Visual Geometry Group) model with setting ImageNet dataset as source data. Firstly, all image data were pre-processed, including random rotation, cut and overturn, etc. Secondly, VGG-16 network was introduced to transfer the features in the convolutional layers, and to connect the fully connected layers network pre-trained by smoke data. Finally, the smoke recognition model was achieved. Experiments were conducted on open datasets and real-world smoke images. The experimental results show that the accuracy of the proposed method is higher than those of current smoke image recognition methods, and the accuracy is more than 96%.

Key words: deep learning, transfer learning, smoke recognition, small dataset

中图分类号: