/YOLOv3_mask_detect

Face mask detection using YOLOv3 on GoogleColab

Primary LanguageJupyter Notebook

YOLOv3-face-mask-detection

Face mask detection using YOLOv3 on Google colab

Photo

image

Video

image

Data

  1. Mask_Dataset

    put Mask_Dataset into yolo directory (/content/darknet/Mask/yolo)

  2. weights

    create weight directory, put weights into weights directory (/content/darknet/weights)

Usage

train

./darknet detector train "/content/darknet/Mask/obj.data" "/content/darknet/Mask/yolov3_mask.cfg" "/content/darknet/weights/darknet53.conv.74" -dont_show

detect photo

./darknet detect Mask/yolov3_mask.cfg weights/yolov3_mask_last.weights Mask/demo/Mask_121.jpg

detect video

./darknet detector demo /content/darknet/Mask/obj.data Mask/yolov3_mask.cfg /content/darknet/weights/yolov3_mask_last.weights -dont_show /content/darknet/Mask/demo/Mask_2.mp4 -i 0 -out_filename predictions.avi

Reference