基于YOLOX轻量化网络的公路监控场景烟雾识别

杜渐, 宋建斌, 胡弘毅, 符锌砂

交通运输研究 ›› 2022, Vol. 8 ›› Issue (4) : 118-125.

交通运输研究 ›› 2022, Vol. 8 ›› Issue (4) : 118-125. DOI: 10.16503/j.cnki.2095-9931.2022.04.011

基于YOLOX轻量化网络的公路监控场景烟雾识别

作者信息 +

Smoke Recognition of Highway Surveillance Scenes Based on YOLOX Light-weight Network

  • DU Jian 1 ,  
  • SONG Jian-bin 1 ,  
  • HU Hong-yi 2 ,  
  • FU Xin-sha 2
Author information +
文章历史 +

摘要

针对公路监控场景下烟雾识别精确度低、传统深层卷积神经网络对计算资源占用量大的问题,基于新型目标识别网络YOLOX构建了公路场景烟雾识别系统。首先收集并标注多公路监控场景下的烟雾图像样本数据,增加Smoke100K部分数据集并整合为公路场景烟雾数据集。同时,搭建并训练多种结构不同层数的YOLOX目标识别网络,利用深度可分离卷积方法对YOLOX网络主体结构进行精简优化,构建nano网络。训练过程中,采用Mosaic数据增强手段对数据集进行扩充,最后导入烟雾图像测试集,通过网络识别精确度、模型大小以及总参数量等指标对4种YOLOX识别网络进行系统测试及对比分析。测试结果表明,YOLOX系列模型均能较好地完成烟雾识别任务,其中nano网络识别精确度达到90.29%,而总参数量仅为m网络的1/25,说明利用数据增强方法提高训练集场景多样性能有效提高网络的泛化能力,在测试场景下提高对烟雾的识别精确度。此外,深度可分离卷积方法能有效压缩模型规模,减少计算量,同时保留网络的特征提取能力。

Abstract

Aiming at the low accuracy of smoke recognition in highway monitoring scene and high hardware device requirements in traditional convolutional neural network, a smoke recognition system in highway scene was constructed based on the new object detection network YOLOX. Firstly, smoke images with Smoke 100K datasets in multi-road monitoring scenes were collected and labeled. Then 4 kinds of YOLOX networks with different layers were built and trained. The depthwise separable convolution was used to simplify and optimize the main structure of YOLOX network, build the nano network. Based on this, Mosaic data enhancement method was used to expand the datasets during training process. Finally, by importing the test datasets, 4 kinds of YOLOX networks with different sizes were systematically tested and compared from the network detection accuracy, model size and total number of parameters. The results show that YOLOX series models can implement the smoke detection task well. The accuracy of nano network is 90.29 % and the total number of parameters is only 1/25 of m network, which means that using data enhancement method to improve the scene diversity of datasets can effectively advance the generalization and recognition accuracy of the network. In addition, depthwise separable convolution can effectively compress the size of the network and reduce the amount of calculation, retaining the feature extraction ability of the network.

关键词

烟雾识别 / 轻量化网络 / 数据增强 / 深度可分离卷积 / 烟雾图像数据集

Key words

smoke recognition / light-weight deep neural network / data augmentation / deep separable convolution / smoke images datasets

引用本文

导出引用
杜渐, 宋建斌, 胡弘毅, . 基于YOLOX轻量化网络的公路监控场景烟雾识别[J]. 交通运输研究. 2022, 8(4): 118-125 https://doi.org/10.16503/j.cnki.2095-9931.2022.04.011
DU Jian, SONG Jian-bin, HU Hong-yi, et al. Smoke Recognition of Highway Surveillance Scenes Based on YOLOX Light-weight Network[J]. Transport Research. 2022, 8(4): 118-125 https://doi.org/10.16503/j.cnki.2095-9931.2022.04.011

参考文献

[1]
交通运输部. 2020年交通运输行业发展统计公报[J]. 交通财会, 2021(6):92-97.
[2]
康娜. 公路隧道运营火灾风险综合评估模式研究[D]. 徐州: 中国矿业大学, 2019.
[3]
胡燕, 王慧琴, 姚太伟, 等. 基于Harris特征点检测与跟踪的火灾烟雾识别[J]. 计算机工程与应用, 2014, 50(21):180-183,194.
[4]
刘鹏. 结合多种图像特征的烟雾识别时序方法[D]. 昆明: 昆明理工大学, 2019.
[5]
袁雯雯, 姜树海, 史晨辉. 基于改进GMM算法的林火烟雾识别研究[J]. 火灾科学, 2019, 28(3):149-155.
[6]
陈超, 齐峰. 卷积神经网络的发展及其在计算机视觉领域中的应用综述[J]. 计算机科学, 2019, 46(3):63-73.
[7]
REN S Q, HE K M, 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.
[8]
REDMON J, DIVVALA S, GIRSHICK R, et al. You Only Look Once: unified, real-time object detection[C]// 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Las Vegas:IEEE, 2016: 779-788.
[9]
杨明潇. 基于深度学习的林火烟雾识别[D]. 北京: 北京林业大学, 2019.
[10]
冯路佳, 王慧琴, 王可, 等. 基于目标区域的卷积神经网络火灾烟雾识别[J]. 激光与光电子学进展, 2020, 57(16):83-91.
[11]
SAPONAR S, ELHANASHI A, GAGLIARDI A. Real-time video fire/smoke detection based on CNN in antifire surveillance systems[J]. Journal of Real-Time Image Processing, 2021, 18(7553): 1-12.
[12]
GE Z, LIU S T, WANG F, et al. YOLOX: Exceeding YOLO Series in 2021[J]. arXiv preprint, 2021: arXiv: 2107. 08430.
[13]
汪辉, 高尚兵, 周君, 等. 基于YOLOv3的多车道车流量统计及车辆跟踪方法[J]. 国外电子测量技术, 2020, 39(2):42-46.
[14]
方卓琳. 基于YOLOv3的道路交通环境行人检测技术研究[D]. 广州: 华南理工大学, 2019.
[15]
申雷霄, 刘军. 基于机器视觉的公路交通标志自动化巡检系统[J]. 交通运输研究, 2018, 4(5):71-76.
[16]
GLENN J. YOLOv5[DB/OL]. (2020-2-10) [2022-2-12]. https://github.com/ultralytics/yolov5.
[17]
BOCHKOVSKIY A, WANG C Y, LIAO H Y M. YOLOv4: Optimal speed and accuracy of object detection[J]. arXiv preprint, 2020: arXiv: 2004.10934.
[18]
SIFRE L, MALLAT S. Rigid-motion scattering for texture classification[J]. Computer Science, 2014, 3559: 501-515.
[19]
CHENG H Y, YIN J L, CHEN B H, et al. Smoke 100k: a database for smoke detection[C]// 2019 IEEE 8th Global Conference on Consumer Electronics (GCCE). Osaka:IEEE, 2019: 596-597.
[20]
LIN T Y, MAIRE M, BELONGIE S, et al. Microsoft COCO: common objects in context[C]// European Conference on Computer Vision. Cham, Germany: Springer, 2014: 740-755.

基金

国家自然科学基金项目(51978283)

Accesses

Citation

Detail

段落导航
相关文章

/