Skip to content

Latest commit

 

History

History
50 lines (32 loc) · 1.1 KB

README.md

File metadata and controls

50 lines (32 loc) · 1.1 KB

Tensorflow face detection

Tensorflow face detection implementation based on Mobilenet SSD V2, trained on Wider face dataset using Tensorflow object detection API.

Dependencies

  • Tensorflow >= 1.12
  • OpenCv
  • imutils
pip install -r requirements.txt

Usage

python face_detection_webcam.py

Docker

docker build -t face_detection .

Run the project with pre-trained model :

bash runDetection.sh

Result

Achieves 19 FPS with a 640x480 resolution on Intel Core i7-7600U CPU 2.80GHz × 4 U.

ex1 ex2

Train Model

If you want to train your own model, i advise you to follow the tutorial about tensorflow object detection api, you'll just need to download an annotated dataset.

Reference