Face Mask Detector
Published:
Trained RetinaNet with ResNet-50 as backbone using Wobot intelligence’s face mask dataset on kaggle. Though the amount of training time was limited by GPU hours on kaggle kernels, the detection of boxes on the test set was impressive.
Data: Wobot intelligence face mask data
Algorithm: RetinaNet, ResNet-50
Major tool(s): Pytorch, OpenCV, Numpy, Matplotlib, Pandas
Code link
Some predictions made by the model:
