실시간 마스크 착용 여부 감지 모델
- Mobilenet V2 CNN 모델을 사용하여 마스크 착용 여부를 Predict함.
- Face Detection은 Caffe2 프레임워크 사용
OpenCV with python3
- keras, plaidml-keras(For AMD GPU Computing)
- numpy
- matplotlib
- argparse
- tensorflow 2.2.0
$ python train_mask_detector.py --dataset dataset
$ python detection.py --image [Path to Image]
- detection.py 내의 video path 변경 후 사용
$ python detection_video.py