Detect face mask in an Image
Project description
SVM facemask classifier
Pretrained Model used: Keras-facenet model () This model is used to detect the faces in a image and create a embedding vector of the faces which would be passed to a SVM classifier. The embedding vector is of dimension 512.
Download the SVM pretrained model https://github.com/Baskar-t/face-mask-classifier/tree/main/models
Class mask_predictor
image - path of the image file to predict the mask
pretrained path of the SVM model
def pred_mask(image,svm_model_path) ...... return img, box_list, label_list, proba_list
img - image vector
box list - list of image box of faces
label_list - list of labels for the predicted face images (Mask , No Mask)
proba_list - list of probability of predicted labels
Usage:
mask_predictor.pred_mask(image path, path_svm_model)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for face_mask_classifier-1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3bc019f0833bbb9fdc570bc69ceb6246ac92b459f8657963bc4b4cc75e3f4c30 |
|
MD5 | 97e385318379952dd4d6fc9659affaf5 |
|
BLAKE2b-256 | 56b182cb1fe98f1da1ea92ed1b06164e593c5faae4aabc7abc338cca3ccfe51e |