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

基于深度迁移学习的烟雾识别方法
引用本文:王文朋,毛文涛,何建樑,窦智. 基于深度迁移学习的烟雾识别方法[J]. 计算机应用, 2017, 37(11): 3176-3181. DOI: 10.11772/j.issn.1001-9081.2017.11.3176
作者姓名:王文朋  毛文涛  何建樑  窦智
作者单位:1. 河南师范大学 计算机与信息工程学院, 河南 新乡 453007;2. 河南省高校计算智能与数据挖掘工程技术中心, 河南 新乡 453007
基金项目:国家自然科学基金资助项目(U1204609);河南省高校科技创新人才支持计划(15HASTIT022);河南省高校青年骨干教师资助计划(2014GGJS-046);河南师范大学优秀青年科学基金资助项目(14YQ007);河南省高等学校重点科研项目计划(15A520078);河南省科技攻关项目(172102210333)。
摘    要:针对传统的基于传感器和图像特征的烟雾识别方法易被外部环境干扰且识别场景单一,从而造成烟雾识别精度较低,而基于深度学习的识别方法对数据量要求较高,对于烟雾数据缺失或数据来源受限的情况模型识别能力较弱的问题,提出一种基于深度迁移学习的烟雾识别方法。将ImageNet数据集作为源数据,利用VGG-16模型进行基于同构数据下的特征迁移。首先,将所有的图像数据进行预处理,对每张图像作随机变换(随机旋转、剪切、翻转等);其次,引入VGG-16网络,将其卷积层特征进行迁移,并连接预先使用烟雾数据在VGG-16网络中训练过的全连接层;进而构建出基于迁移学习的深度网络,从而训练得到烟雾识别模型。利用公开数据集以及真实场景烟雾图像进行实验验证,实验结果表明,和现有主流烟雾图像识别方法相比,所提方法有较高的烟雾识别率,实验精度达96%以上。

关 键 词:深度学习  迁移学习  烟雾识别  微量数据集  
收稿时间:2017-05-16
修稿时间:2017-06-07

Smoke recognition based on deep transfer learning
WANG Wenpeng,MAO Wentao,HE Jianliang,DOU Zhi. Smoke recognition based on deep transfer learning[J]. Journal of Computer Applications, 2017, 37(11): 3176-3181. DOI: 10.11772/j.issn.1001-9081.2017.11.3176
Authors:WANG Wenpeng  MAO Wentao  HE Jianliang  DOU Zhi
Affiliation: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
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%.
Keywords:deep learning   transfer learning   smoke recognition   small dataset
点击此处可从《计算机应用》浏览原始摘要信息
点击此处可从《计算机应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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