7123 Deep Learning final project
Author: Pengyu Zhang([email protected]), Xiao Xie([email protected]), Taiyan Yu([email protected])
Download dataset from https://www.kaggle.com/datasets/andrewmvd/face-mask-detection, unzip it under this directory and rename "archive" to "input".
Used as baseline comparison
Author: https://www.kaggle.com/code/daniel601/pytorch-fasterrcnn/notebook
Training script: Mask_scoring_RCNN.py
Run the Mask_detection_yolov5 noteboook
Here are some of the blog posts and codes that we have used as references:
[1] https://www.kaggle.com/code/daniel601/pytorch-fasterrcnn/notebook
[2] https://github.com/ultralytics/YOLOv5
[3] https://blog.roboflow.com/yolov4-versus-yolov5/
[4] https://www.kaggle.com/code/sandhyakrishnan02/face-mask-detection-using-pytorch
[5] https://www.kaggle.com/code/chandlertimm/face-mask-detection-with-empirical-attention