장면 복잡도 기반 적응적 얼굴 마스크 탐지 모델
Adaptive Face Mask Detection System based on Scene Complexity Analysis
1
조회
0
다운로드
2
인용
0
참고문헌
키워드
초록
코로나바이러스-19(COVID-19)의 대유행에 따라 전 세계 수많은 확진자가 발생하고 있으며 국민을 불안에 떨게 하고 있다. 바이러스 감염 확산을 방지하기 위해서는 마스크를 제대로 착용하는것이 필수적이지만 몇몇 사람들은 마스크를 쓰지 않거나 제대로 착용하지 않고 있다. 본 논문에서는 영상 이미지에서의 효율적인 마스크 감지 시스템을 제안한다. 제안 방법은 우선 입력 이미지의 모든 얼굴의 영역을 YOLOv5를 사용하여 감지하고 감지된 얼굴의 수에 따라 3가지의 장면복잡도(Simple, Moderate, Complex) 중 하나로 분류한다. 그 후 장면 복잡도에 따라 3가지ResNet(ResNet-18, 50, 101) 중 하나를 기반으로 한 Faster-RCNN을 사용하여 얼굴 부위를 감지하고마스크를 제대로 착용하였는지 식별한다. 공개 마스크 감지 데이터셋을 활용하여 실험한 결과 제안한 장면 복잡도 기반 적응적인 모델이 다른 모델에 비해 가장 성능이 뛰어남을 확인하였다.
Abstract
Coronavirus disease 2019 (COVID-19) has affected the world seriously. Every person is required for wearing a mask properly in a public area to prevent spreading the virus. However, many people are not wearing a mask properly. In this paper, we propose an efficient mask detection system. In our proposed system, we first detect the faces of input images using YOLOv5 and classify them as the one of three scene complexity classes (Simple, Moderate, and Complex) based on the number of detected faces. After that, the image is fed into the Faster-RCNN with the one of three ResNet (ResNet-18, 50, and 101) as backbone network depending on the scene complexity for detecting the face area and identifying whether the person is wearing the mask properly or not. We evaluated our proposed system using public mask detection datasets. The results show that our proposed system outperforms other models.
참고문헌 (17)
- A. Krizhevsky, "Imagenet classification with deep convolutional neural networks", Advances in Neural Information Processing Systems, 25, pp.1097~1105 (2012)
- O. Russakovsky, "Imagenet large scale visual recognition challenge", International Journal of Computer Vision, 115, pp.211~252 (2015)
- C. Szegedy, "Going deeper with convolutions", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.1~9 (2015)
- K. Simonyan, "Very deep convolutional networks for large-scale image recognition", arXiv:1409.1556, ~ (2014)
- K. He, "Deep residual learning for image recognition", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.770~778 (2016)
- J. Redmon, "You only look once: Unified, real-time object detection", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.779~788 (2016)
- R. Girshick, "Rich feature hierarchies for accurate object detection and semantic segmentation", Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp.580~587 (2014)
- R. Girshick, "Fast R-CNN", Proceedings of the IEEE International Conference on Computer Vision, pp.1440~1448 (2015)
- N. Tajbakhsh, "Convolutional neural networks for medical image analysis : Full training or fine tuning?", IEEE Transactions on Medical Imaging, 35(5), pp.1299~1312 (2016)
- S. J. Pan, "A survey on transfer learning", IEEE Transactions on Knowledge and Data Engineering, 22(10), pp.1345~1359 (2009)
- "Mask dataset"[URL]
- P. Viola, "Rapid object detection using a boosted cascade of simple features", Proceedings of the 2001 IEEE Computer Society Conference on Computer Vision and Pattern Recognition. CVPR 2001, 1 (2001), IEEE
- S. Akçay, "Transfer learning using convolutional neural networks for object classification within x-ray baggage security imagery", 2016IEEE International Conference on Image Processing(ICIP), pp.1057~1061 (2016), IEEE
- H. Yang, "Transfer-learning-based online Mura defect classification", IEEE Transactions on Semiconductor Manufacturing, 31(1), pp.116~123 (2018)
- I. M. Baltruschat, "Comparison of deep learning approaches for multi-label chest X-ray classification", Scientific Reports, 9(1), pp.1~10 (2019)
- S. Christodoulidis, "Multisource transfer learning with convolutional neural networks for lung pattern analysis", IEEE Journal of Biomedical and Health Informatics, 21(1), pp.76~84 (2016)
- J. Kang, "Ensemble of instance segmentation models for polyp segmentation in colonoscopy images", IEEE Access, 7, pp.26440~26447 (2019)