Open AccessDB-Net: Dual Attention Network with Bilinear Pooling for Fire-Smoke Image Classification
Author(s) -
You Chen,
Zhen Li,
Meng Li,
Zongliang Gao,
Wei Li
Publication year - 2020
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/1631/1/012054
Subject(s) - pooling , bilinear interpolation , computer science , merge (version control) , artificial intelligence , smoke , pattern recognition (psychology) , contextual image classification , dual (grammatical number) , machine learning , image (mathematics) , computer vision , engineering , information retrieval , art , literature , waste management
In recent years, with the help of deep learning, image classification has been significantly improved. More and more researchers are seeking fire alarm solutions based on these artificial intelligence algorithms. However, this raises new challenges in an effective method of classification for fire-smoke images. In this paper, we propose a Dual Attention Network with Bilinear Pooling of three steps to accomplish the task of fire-smoke classification. We first extract features of fire pictures or smoke ones by using ResNet-50 as a basic net. Then we attach two kinds of attention modules, channel attention and spatial attention, which are called dual attention, to extract key information ‘what’ and ‘where’ from pictures. Finally, we merge them by using the bilinear pooling module which has been shown to be effective at improving our classification rate. Results show that our most accurate model can reach 90.11% per-image accuracy, which is improved by 4.81% compared to the traditional ResNet-50.