Skip to content

A faster rcnn implementation of face detector. Use WIDER FACE dataset, and tensorflow.

Notifications You must be signed in to change notification settings

Novelfor/FaceDetector_tf

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Tensorflow Face Detector

It's a ResNet50, Faster R-CNN based face detector. It include all steps for training a faster rcnn in a single jupyter notebook.

Training on WIDER FACE Dataset.

The jupyter includes the code for preparing data from WIDER FACE datasets. Download the data from WIDER FACE and put them in data folder.

Model details.

We train the model tranfered from the ImageNet pretrained ResNet50. The download code is in jupyter notebook. Use the original image withou image resizing. And use some image augmentations to improve model performance. Use learning rate decay. Use image resize instead of ROI Pooling, it's a common implementation of ROI Pooling in tensorflow. (It's different with the oral paper.) Use Joint Training for ROI and RPN.

Some Results.

Examples

Something else.

For Face Detection, R-CNN base method may not be a best way to implement it. MTCNN, Cascade CNN may be better. No implementation of multiple GPUS.

About

A faster rcnn implementation of face detector. Use WIDER FACE dataset, and tensorflow.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published