/Face_Recognize

ArcFace pytorch

Primary LanguageJupyter Notebook

Face_recognize_system

Requirements

Python 3.5+

Commanline:

pip3 install -r requirements.txt

Usage:

Download

git clone https://github.com/vanlong96tg/Face_recognize_pytorch
mkdir face_recognize/weights
cd face_recognize/weights
wget https://www.dropbox.com/s/akktsgxp0n8cwn2/model_mobilefacenet.pth?dl=0 -O model_mobilefacenet.pth
wget https://www.dropbox.com/s/kzo52d9neybjxsb/model_ir_se50.pth?dl=0 -O model_ir_se50.pth
wget https://www.dropbox.com/s/rxavczg9dlxy3a8/model_ir50.pth?dl=0 -O model_ir50.pth

Python:

Run demo:

cd demo
python infer_on_video.py 

Run web_demo:

cd web
python app.py

Run demo system manager for rasperi:

cd stream
python client.py -s ip_adress
python servr.py

Trainning:

Pretrain:

*Check in folder 'src'.

Acknowledgement