goktugyildirim / RealTimeMedicalMaskDetection

Real Time Medical Mask Detection using OpenCV and PyTorch
1 stars 0 forks source link

RealTimeMedicalMaskDetection

Real Time Medical Mask Detection using OpenCV and PyTorch:

This repostory contains two IPython Notebook file:

1) Masked_Unmasked_Classifier: I developed a PyTorch model that can classify if the face carries the medical mask or not, by performing a transfer learning concept on the pre-trained ResNet-50 model. This model was tested with 576 different pictures and achieved %99 accuracy.

2) Masked_Unmasked_Detection: This file performs real-time detection by using a pre-trained Haar Cascade model from the OpenCV module and PyTorch model that was developed by me.

Demonstration: https://www.youtube.com/watch?v=fwDPIu_rsrs&feature=youtu.be

Download face detection and classifier models: https://drive.google.com/open?id=1d4r5o7dr965-W173MfBhuhYUt6qCE7bE