Journal of Xidian University ›› 2019, Vol. 46 ›› Issue (6): 52-59.doi: 10.19665/j.issn1001-2400.2019.06.008

Previous Articles     Next Articles

Algorithm for segmentation of smoke using the improved DeeplabV3 network

WANG Ziyi1,SU Yuting1,LIU Yanyan2,ZHANG Wei3   

  1. 1. School of Electrical Automation and Information Engineering, Tianjin University, Tianjin 300072, China
    2. School of Electronic Information and Optical Engineering, Nankai University, Tianjin 300071, China
    3. School of Microelectronics, Tianjin University, Tianjin 300072, China
  • Received:2019-06-07 Online:2019-12-20 Published:2019-12-21

Abstract:

Existing smoke detection methods depend mostly on features which are selected manually and the smoke areas in video images often cannot be segmented accurately. This paper proposes an improved DeeplabV3 smoke segmentation algorithm based on this. A feature refinement module is added after the basic encoder network to weaken the gridding effects caused by dilated convolutions. For the non-rigid objects such as smoke with variable scales and postures, the Atrous Spatial Pyramid Pooling module is combined with the deformable convolution to better adapt to the smoke deformation. And a channel attention decoder module is proposed to further restore the spatial details of smoke images. According to the test of the smoke image data set, the proposed model has a faster average prediction speed of 71.73ms per image. Besides, compared with the DeeplabV3 network, this algorithm can lead to a higher MPA (Mean Pixel Accuracy) of 97.78% and a higher MIoU (Mean Intersection over Union) of 91.21%, thus improving the performance by 0.56% and 2.17% respectively, and making it more suitable for smoke segmentation. Public smoke video test results show that this model outperforms other video-based smoke detection methods for the detection rate, and that it is of certain research significance and practical value.

Key words: image processing, smoke detection, semantic segmentation, deformable convolution, attention mechanism, deep learning

CLC Number: 

  • TP391

Baidu
map