Progressive Calibration Networks (PCN) is an accurate rotation-invariant face detector running at real-time speed on CPU. This is an implementation for PCN.
Some detection results can be viewed in the following illustrations:
PCN is designed aiming for low time-cost. We compare PCN's speed with other rotation-invariant face detectors' on standard VGA images(640x480) with 40x40 minimum face size. The detectors run on a desktop computer with 3.4GHz CPU, GTX Titan X. The speed results together with the recall rate at 100 false positives on multi-oriented FDDB are shown in the following table. Detailed experiment settings can be found in our paper. It is worth mentioning that converting the square results to rectangles or ellipses is helpful to fit the ground-truth data. In this way, better accuracy can be achieved. But we do not convert the results here.
Set minimum size of faces to detect (size
>= 20)
detector.SetMinFaceSize(size);
Set scaling factor of image pyramid (1.4 <= factor
<= 1.6)
detector.SetImagePyramidScaleFactor(factor);
Set score threshold of detected faces (0 <= thresh1
, thresh2
, thresh3
<= 1)
detector.SetScoreThresh(thresh1, thresh2, thresh3);
Smooth the face boxes or not (smooth = true or false, recommend using it on video to get stabler face boxes)
detector.SetVideoSmooth(smooth);
See picture.cpp and video.cpp for details. If you want to reproduce the results on FDDB, please run fddb.cpp. You can rotate the images in FDDB to get FDDB-left, FDDB-right, and FDDB-down, then test PCN on them respectively.
- Linux
- Caffe
- OpenCV (2.4.10, or other compatible version)
Build the library
make
Run the demo
sh run.sh picture/crop/video/fddb
For python interface (only supported in Ubuntu 18.04):
sudo apt install libcaffe-cpu-dev
make
sudo make install
python3 PyPCN.py
- Python3 interface (by EmilWine)
- PyCaffe (by daixiangzi)
- PyTorch (by siriusdemon)
- IOS (by elhoangvu)
- NCNN (by HandsomeHans)
- OpenCV (by richipower)
- Windows (by imistyrain)
- ETOS (by etosworld)
-
How to get faces with different rotation-in-plane angles before training?
If you find PCN useful in your research, please consider citing:
@inproceedings{PCN_CVPR18,
title = {Real-Time Rotation-Invariant Face Detection with Progressive Calibration Networks},
author = {Xuepeng Shi and Shiguang Shan and Meina Kan and Shuzhe Wu and Xilin Chen},
booktitle = {CVPR},
year = {2018},
}