Skip to content

JavisPeng/FaceRecognitionWithMask

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

python enviroment

dlib==19.20.0
opencv_python==4.2.0.34
onnxruntime==1.2.0
fastapi==0.61.0
numpy==1.19.1

How to use ?

1. build the target database

Put all target pictures in a directory (data/mask_nomask in this project), one person with some of pictures (masked and none-masked is recommended), the way of file naming is name.index.jpg

target database

2. dowload the weight files

you can download the all weight files from

  1. baiduyunpan passwd:60ri (need a baidu account)
  2. lanzhoucloud (no restriction to download)

and put them in data directory

3. test a video demo

python demo_video.py  --face_db_root data/mask_nomask --input_video_path 0.mp4 --output_video_path output.mp4

face_video_demo

4. test in fastapi server

uvicorn main:app --reload --host 0.0.0.0

you can upload a picture from http://127.0.0.1:8000/docs, and retrive the face recognition information [name,box,mask or not] fastapi

blogs

https://blog.csdn.net/jiangpeng59/article/details/107986046

reference

https://github.com/sgrvinod/a-PyTorch-Tutorial-to-Object-Detection https://github.com/AIZOOTech/FaceMaskDetection https://github.com/davisking/dlib

About

Quickly build a server to face recognition with mask

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages