Abstract
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.
Published Version
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.