Skip to content

open-tube/Face-detection-with-mobilenet-ssd

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

46 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Description

This is a implementation of mobilenet-ssd for face detection written by keras, which is the first step of my FaceID system. You can find another two repositories as follows:

  1. Face-detection-with-mobilenet-ssd
  2. Face-Alignment-with-simple-cnn
  3. Face-identification-with-cnn-triplet-loss

prepare data

  1. You are advised to use CUDA-compatible GPUs to train the model.
  2. Download WIDER Face from Official Website , and put it into data_path folder in face_train.ipynb.
  3. wider_extract.py has been modified to show the method of exctracting faces from the datasets. It's easy to follow.

train

  1. Follow face_train.ipynb step by step. You can change the parameters for better performance.
  2. wider_train_small.npy and wider_val_small.npy are made to testing the network. If you don't have enough gpu resources, you can also use them for training.

test

Here are some testing results. It seems good but improvement is still needed. For example, the Bbox is a little bit inaccurate.

to do

  1. Evaluation is on the way.
  2. MobileNetV2 version.

License

MIT LICENSE

References

  1. rykov8/ssd_keras
  2. rcmalli/keras-mobilenet

About

Face detection with mobilenet-ssd written by Keras.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 54.2%
  • Python 45.8%