计算机科学 ›› 2019, Vol. 46 ›› Issue (6): 301-304.doi: 10.11896/j.issn.1002-137X.2019.06.045

• 图形图像与模式识别 • 上一篇    下一篇

基于YOLOv2的视频火焰检测方法

杜晨锡1,2, 严云洋1,2, 刘以安1, 高尚兵2   

  1. (江南大学物联网工程学院 江苏 无锡214122)1
    (淮阴工学院计算机与软件工程学院 江苏 淮安223003)2
  • 收稿日期:2018-06-01 发布日期:2019-06-24
  • 通讯作者: 严云洋(1967-),男,博士,教授,CCF会员,主要研究方向为数字图像处理、模式识别,E-mail:yunyang@hyit.edu.cn
  • 作者简介:杜晨锡(1991-),男,硕士生,主要研究方向为机器学习、模式识别,E-mail:tencyrush@163.com;刘以安(1963-),男,博士,教授,主要研究方向为模式识别、数据融合;高尚兵(1981-),男,博士,副教授,主要研究方向为数字图像处理、模式识别。
  • 基金资助:
    国家自然科学基金项目(61402192),江苏省“六大人才高峰”项目(2013DZXX-023),江苏省“青蓝工程”,江苏省高等学校自然科学研究重大项目(18KJA520001),淮安市“533英才工程”资助。

Video Fire Detection Method Based on YOLOv2

DU Chen-xi1,2, YAN Yun-yang1,2, LIU Yi-an1, GAO Shang-bing2   

  1. (School of Internet of Things Engineering,Jiangnan University,Wuxi,Jiangsu 214122,China)1
    (Faculty of Computer & Software Engineering,Huaiyin Institute of Technology,Huaian,Jiangsu 223003,China)2
  • Received:2018-06-01 Published:2019-06-24

摘要: 一般火焰检测方法由于对复杂场景的应变能力较差,因此检测率较低。文中提出了一种基于改进的YOLOv2网络的深度学习火焰检测方法,来自动提取火焰特征;同时,针对特征提取过程中信息丢失的问题,采用聚类选取候选框,以多尺度特征融合的方法融合高层与浅层特征信息,进一步提高了模型的检测率。在Bilkent大学火焰视频数据集上的实验结果表明,该方法的平均正检率达到了98.8%,检测速率达到40帧/s,具有较强的鲁棒性和实时性。

关键词: YOLOv2, 多级特征, 火焰检测, 聚类, 特征融合

Abstract: It is difficult for general flame detection methods to adapt to complex scenes,so the detection rates is low.This paper proposed a deep learning flame detection method based on an improved YOLOv2 network to extract the flame features automatically.In order to avoid the information loss in the feature extraction process,the selected anchor box by clustering is suggested and multi-scale feature fusion method is used to fuse high-level and shallow feature information,to further improve the detection rate of the model.Experimental results on the Bilkent University flame video dataset show that the average true inspection rate of the proposed method is 98.8%,and the detection rate is 40 frames/s,so its robustness and real-time performance are strong.

Key words: Clustering, Feature fusion, Fire detection, Multi-scale feature, YOLOv2

中图分类号: 

  • TP391
[1]DIMITROPOULOS K,BARMPOUTIS P,GRAMMALIDIS N.Spatio-Temporal Flame Modeling and Dynamic Texture Analysis for Automatic Video-Based Fire Detection[J].IEEE Tran-sactions on Circuits & Systems for Video Technology,2015,25(2):339-351.
[2]CELIK T,DEMIREL H,OZKARAMANLI H.Automatic fire detection in video sequences[C]∥European Signal Processing Conference.IEEE,2015:1-5.
[3]POOBALAN K,LIEW S C.Fire detection algorithm using ima-ge processing techniques[C]∥International Conference on Artificial Intelligence and Computer Science.2015.
[4]GENG Q T,YU F H,ZHAO H W,et al.New algorithm of flame detection based on color features[J].Journal of Jilin University(Engineering and Technology Edition),2014,44(6):1787-1792.(in Chinese)
耿庆田,于繁华,赵宏伟,等.基于颜色特征的火焰检测新算法[J].吉林大学学报(工学版),2014,44(6):1787-1792.
[5]FRIZZI S,KAABI R,BOUCHOUICHA M,et al.Convolutional neural network for video fire and smoke detection[C]∥Confe-rence of the IEEE Industrial Electronics Society.IEEE,2016:877-882.
[6]MUHAMMAD K,AHMAD J,MEHMOOD I,et al.Convolu-tionalNeural Networks based Fire Detection in Surveillance Vi-deos[J].IEEE Access,2018,PP(99):1.
[7]ZHANG H,WANG K F,WANG F Y.Advances and Perspectives on Applications of Deep Learning in Visual Object Detection[J].Acta Automatic Sinica,2017,43(8):1289-1305.(in Chinese)
张慧,王坤峰,王飞跃.深度学习在目标视觉检测中的应用进展与展望[J].自动化学报,2017,43(8):1289-1305.
[8]REDMON J,FARHADI A.YOLO9000:Better,faster,stronger[C]∥2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).Washington:IEEE Computer Society,2017:6517-6525.
[9]REN S,GIRSHICK R,GIRSHICK R,et al.Faster R-CNN:Towards Real-Time Object Detection with Region Proposal Networks[J].IEEE Transactions on Pattern Analysis & Machine Intelligence,2017,39(6):1137-1149.
[10] HE K,ZHANG X,REN S,et al.Identity Mappings in Deep Residual Networks[C]∥Computer Vision-ECCV 2016.Springer International Publishing,2016:630-645.
[11] PINHEIRO P O,LIN T Y,COLLOBERT R,et al.Learning to refine object segments[C]∥Lecture Notes in Computer Scien-ce.Heidelberg.Springer,2016:75-91.
[12] LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single Shot MultiBox Detector[C]∥European Conference on Computer Vision.Springer,Cham,2016:21-37.
[13]HOWARD A G,ZHU M,CHEN B,et al.MobileNets:Efficient Convolutional Neural Networks for Mobile Vision Applications[J].arXiv preprint arXiv:1704.04861,2017.
[1] 柴慧敏, 张勇, 方敏.
基于特征相似度聚类的空中目标分群方法
Aerial Target Grouping Method Based on Feature Similarity Clustering
计算机科学, 2022, 49(9): 70-75. https://doi.org/10.11896/jsjkx.210800203
[2] 鲁晨阳, 邓苏, 马武彬, 吴亚辉, 周浩浩.
基于分层抽样优化的面向异构客户端的联邦学习
Federated Learning Based on Stratified Sampling Optimization for Heterogeneous Clients
计算机科学, 2022, 49(9): 183-193. https://doi.org/10.11896/jsjkx.220500263
[3] 张颖涛, 张杰, 张睿, 张文强.
全局信息引导的真实图像风格迁移
Photorealistic Style Transfer Guided by Global Information
计算机科学, 2022, 49(7): 100-105. https://doi.org/10.11896/jsjkx.210600036
[4] 程成, 降爱莲.
基于多路径特征提取的实时语义分割方法
Real-time Semantic Segmentation Method Based on Multi-path Feature Extraction
计算机科学, 2022, 49(7): 120-126. https://doi.org/10.11896/jsjkx.210500157
[5] 刘丽, 李仁发.
医疗CPS协作网络控制策略优化
Control Strategy Optimization of Medical CPS Cooperative Network
计算机科学, 2022, 49(6A): 39-43. https://doi.org/10.11896/jsjkx.210300230
[6] 鲁晨阳, 邓苏, 马武彬, 吴亚辉, 周浩浩.
基于DBSCAN聚类的集群联邦学习方法
Clustered Federated Learning Methods Based on DBSCAN Clustering
计算机科学, 2022, 49(6A): 232-237. https://doi.org/10.11896/jsjkx.211100059
[7] 郁舒昊, 周辉, 叶春杨, 王太正.
SDFA:基于多特征融合的船舶轨迹聚类方法研究
SDFA:Study on Ship Trajectory Clustering Method Based on Multi-feature Fusion
计算机科学, 2022, 49(6A): 256-260. https://doi.org/10.11896/jsjkx.211100253
[8] 毛森林, 夏镇, 耿新宇, 陈剑辉, 蒋宏霞.
基于密度敏感距离和模糊划分的改进FCM算法
FCM Algorithm Based on Density Sensitive Distance and Fuzzy Partition
计算机科学, 2022, 49(6A): 285-290. https://doi.org/10.11896/jsjkx.210700042
[9] 陈景年.
一种适于多分类问题的支持向量机加速方法
Acceleration of SVM for Multi-class Classification
计算机科学, 2022, 49(6A): 297-300. https://doi.org/10.11896/jsjkx.210400149
[10] 杨玥, 冯涛, 梁虹, 杨扬.
融合交叉注意力机制的图像任意风格迁移
Image Arbitrary Style Transfer via Criss-cross Attention
计算机科学, 2022, 49(6A): 345-352. https://doi.org/10.11896/jsjkx.210700236
[11] 陈永平, 朱建清, 谢懿, 吴含笑, 曾焕强.
基于外接圆半径差损失的实时安全帽检测算法
Real-time Helmet Detection Algorithm Based on Circumcircle Radius Difference Loss
计算机科学, 2022, 49(6A): 424-428. https://doi.org/10.11896/jsjkx.220100252
[12] 孙洁琪, 李亚峰, 张文博, 刘鹏辉.
基于离散小波变换的双域特征融合深度卷积神经网络
Dual-field Feature Fusion Deep Convolutional Neural Network Based on Discrete Wavelet Transformation
计算机科学, 2022, 49(6A): 434-440. https://doi.org/10.11896/jsjkx.210900199
[13] 蓝凌翔, 池明旻.
基于特征注意力融合网络的遥感变化检测研究
Remote Sensing Change Detection Based on Feature Fusion and Attention Network
计算机科学, 2022, 49(6): 193-198. https://doi.org/10.11896/jsjkx.210500058
[14] 陈佳舟, 赵熠波, 徐阳辉, 马骥, 金灵枫, 秦绪佳.
三维城市场景中的小物体检测
Small Object Detection in 3D Urban Scenes
计算机科学, 2022, 49(6): 238-244. https://doi.org/10.11896/jsjkx.210400174
[15] 邢云冰, 龙广玉, 胡春雨, 忽丽莎.
基于SVM的类别增量人体活动识别方法
Human Activity Recognition Method Based on Class Increment SVM
计算机科学, 2022, 49(5): 78-83. https://doi.org/10.11896/jsjkx.210400024
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!