Gas mask wearing detection based on Faster R-CNN

Article type: Research Article

Authors: Wang, Bangrong | Wang, Jun^{; *} | Xu, Xiaofeng | Bao, Xianglin

Affiliations: School of Computer and Information, Anhui Polytechnic University, Wuhu, China

Correspondence: [*] Corresponding author. E-mail: [email protected].

Abstract: Gas masks are essential respiratory protective equipment commonly used by laborers who work in harsh environments. However, respiratory diseases and accidents can occur due to the absence of gas masks. To prevent these accidents, this paper developed an object detector that uses convolutional neural networks (CNNs) to detect whether workers are wearing gas masks. To achieve this goal, a gas mask detection dataset was constructed derived from real industrial scenarios and Faster R-CNN was improved for gas mask wearing detection. Firstly, to address the multi-scale problem in real scenes, the Feature Pyramid Network was introduced into Faster R-CNN to effectively fuse features between different levels and improve the detection ability of small objects. Secondly, the Online Hard Sample Mining algorithm was used to alleviate the class imbalance problems in the dataset. Finally, Mixup and Mosaic were used in the training process to augment the data and make the model better adapt to different scenes and complex backgrounds. After multiple experiments, the combination of the three optimization strategies improved the mAP0.5:0.95 by 23.2%. This work is an initial attempt at gas mask wearing detection and there is still much room for improvement in terms of model and dataset.

Keywords: Gas mask wearing detection, convolutional neural networks, Faster R-CNN, Feature Pyramid Network, online hard example mining

DOI: 10.3233/AIS-220460

Journal: Journal of Ambient Intelligence and Smart Environments, vol. 16, no. 1, pp. 57-71, 2024