Detect mask, use yolov4 and keras

Development Process

• 1. Use official darknet to train (YOLOv4: https://github.com/AlexeyAB/darknet/)
• 2. Convert to Tensorflow model & save as .pb files (Officail TF model: https://github.com/hunglc007/tensorflow-yolov4-tflite)
• 3. Use Keras load model and use it

Requirements: Tensorflow 2.5:arrow_up: , opencv 4.5.3

Optional: freetype

How to use my python scripts

Download My .h5 file

Run python file on command line

# detect with your webcam
python detect_webcam.py

# if you have freetype module
detect_chinese.py

# detect single image (freetype)
python detect_image.py --input=" your image path " --output=" output path " 

# detect video and save it (freetype)
python detect_video.py --input=" your video path " --output=" output path "

If you want to build opencv with freetype module, check this

How do I train my model?

Download dataset

• I train my model on colab, this is my Code.
  
• Convert to tensorflow, this is my Code

train history