Skip to content

This Python script is used to detect whether a people in the given image are wearing a mask or not. The custom dataset was trained using Darknet.

Notifications You must be signed in to change notification settings

rushad7/mask-detection

Repository files navigation

mask-detection

This Python script is used to detect whether a people in the given image are wearing a mask or not. The custom dataset was trained using Darknet.

Usage for live detection:

python yolo-live-cv2.py --yolo yolo

Usage for detection from images:

python custom_detector.py --image "/path/to/image" --config "/cfg/yolov3-custom.cfg" --weights "weights/yolov3-custom_final.weights" --names "class_names/voc.names"

Misc Files

data_augmentation.py is a Python script to increase the sise of your dataset use data augmentation.
rename.py is to rename the images and annotated .txt files to the same name according to the correct naming rules.
test_train_split.py is to generate the test and train .txt files

About

This Python script is used to detect whether a people in the given image are wearing a mask or not. The custom dataset was trained using Darknet.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages