Skip to content

Face Mask Detection

Face mask detection merupakan sistem yang dapat mendeteksi pemakaian masker pada wajah. Sistem ini dapat membedakan antara wajah bermasker, wajah tidak bermasker, dan jika pemakaian masker salah. Terdapat tiga implementasi yang dibuat dari sistem ini, sebagai berikut:

Based on Haar Cascade

Haar-Cascade merupakan algoritma pendeteksi objek yang menggunakan fitur Haar. Pertama kali dipublikasikan pada 20011. Untuk dapat mendeteksi wajah bermasker dibutuhkan dua buah model: (1) model pendeteksi wajah, (2) model pendeteksi hidung. Kedua model tersebut dapat diperoleh pada repository official opencv. Hasil inference kedua model tersebut dapat dilihat pada gambar dibawah. Implementasi kode dapat dilihat pada Notebook Google Colab.

unmask unmask
Tidak Bermasker Bermasker

Based on OpenCV + MobileNet

Implementasi kedua berbasis OpenCV yang digunakan untuk mendeteksi wajah, setelahnya wajah yang terdeteksi akan diklasifikasikan menggunakan model MobileNetv2. Model MobileNetv2 sebelumnya sudah dilatih pada dataset wajah bermasker dan tidak bermasker. Dataset tersebut dapat diakses pada Dataset Kaggle2. Model pendeteksi wajah OpenCV dapat diakses pada Dataset Kaggle penulis. Sedangkan, proses training dan pretrained model klasifikasi MobileNetv2 dapat diakses pada Notebook Kaggle penulis.

MobileNet

Based on SSD MobileNet

Implementasi ketiga adalah berbasis SSD (Single Shot Detector) yang menggunakan backbone model MobileNetv2 Quantized. Implementasi ini memang saya khususkan untuk penggunaan pada smartphone (Aplikasi Android). Dataset yang digunakan merupakan dataset berformat Pascal VOC yang dapat diakses pada Dataset Kaggle3. Proses training, kuantisasi model dan pembuatan metadata (diperlukan saat pembuatan aplikasi) dapat diakses pada Notebook Kaggle penulis. Repository model dan aplikasi android dapat diakses masing-masing pada model dan aplikasi.

Android


  1. Viola, Paul, and Michael Jones. "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. Vol. 1. Ieee, 2001. 

  2. https://www.kaggle.com/spandanpatnaik09/face-mask-detectormask-not-mask-incorrect-mask 

  3. https://www.kaggle.com/andrewmvd/face-mask-detection