├── .gitignore ├── README.md ├── doc └── TRAIN.md ├── mtcnn ├── __init__.py ├── datasets │ ├── CelebA │ │ └── .gitignore │ ├── FacePoint │ │ └── .gitignore │ ├── WIDER_FACE │ │ └── .gitignore │ ├── __init__.py │ ├── celeba.py │ └── wider_face.py ├── deploy │ ├── __init__.py │ ├── align.py │ ├── batch_detect.py │ ├── detect.py │ ├── models │ │ ├── onet.npy │ │ ├── pnet.npy │ │ └── rnet.npy │ └── tracker.py ├── network │ ├── __init__.py │ └── mtcnn_pytorch.py ├── train │ ├── __init__.py │ ├── data.py │ ├── gen_landmark.py │ ├── gen_onet_train.py │ ├── gen_pnet_train.py │ ├── gen_rnet_train.py │ └── train_net.py └── utils │ ├── __init__.py │ ├── align_trans.py │ ├── cp2tform.py │ ├── draw.py │ ├── functional.py │ └── nms │ ├── __init__.py │ ├── cpu_nms.c │ ├── cpu_nms.pyx │ ├── gpu_nms.cpp │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── nms_kernel.cu │ └── py_cpu_nms.py ├── output └── caffe_models │ ├── det1.caffemodel │ ├── det1.prototxt │ ├── det2.caffemodel │ ├── det2.prototxt │ ├── det3.caffemodel │ ├── det3.prototxt │ ├── det4.caffemodel │ └── det4.prototxt ├── scripts ├── convert_caffe_model.py ├── detect_on_video.py ├── gen_onet_train.py ├── gen_pnet_train.py ├── gen_rnet_train.py ├── track_on_video.py ├── train_onet.py ├── train_pnet.py └── train_rnet.py ├── setup.py ├── tests ├── __init__.py ├── asset │ ├── images │ │ ├── audrey.jpg │ │ ├── bksomels.jpg │ │ ├── gpripe.jpg │ │ ├── office5.jpg │ │ └── roate.jpg │ └── video │ │ └── track.mp4 ├── test_align.py ├── test_batch_detection.py ├── test_data.py ├── test_datasets.py ├── test_detection.py ├── test_functional.py ├── test_gptd.py ├── test_landmarks.py ├── test_net_jit.py ├── test_net_pytorch.py ├── test_tracker.py └── test_train.py └── tutorial ├── README.md ├── detect_step_by_step.ipynb └── face_align.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/README.md -------------------------------------------------------------------------------- /doc/TRAIN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/doc/TRAIN.md -------------------------------------------------------------------------------- /mtcnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/__init__.py -------------------------------------------------------------------------------- /mtcnn/datasets/CelebA/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /mtcnn/datasets/FacePoint/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /mtcnn/datasets/WIDER_FACE/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /mtcnn/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/datasets/__init__.py -------------------------------------------------------------------------------- /mtcnn/datasets/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/datasets/celeba.py -------------------------------------------------------------------------------- /mtcnn/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/datasets/wider_face.py -------------------------------------------------------------------------------- /mtcnn/deploy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/__init__.py -------------------------------------------------------------------------------- /mtcnn/deploy/align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/align.py -------------------------------------------------------------------------------- /mtcnn/deploy/batch_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/batch_detect.py -------------------------------------------------------------------------------- /mtcnn/deploy/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/detect.py -------------------------------------------------------------------------------- /mtcnn/deploy/models/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/models/onet.npy -------------------------------------------------------------------------------- /mtcnn/deploy/models/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/models/pnet.npy -------------------------------------------------------------------------------- /mtcnn/deploy/models/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/models/rnet.npy -------------------------------------------------------------------------------- /mtcnn/deploy/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/deploy/tracker.py -------------------------------------------------------------------------------- /mtcnn/network/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mtcnn/network/mtcnn_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/network/mtcnn_pytorch.py -------------------------------------------------------------------------------- /mtcnn/train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mtcnn/train/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/train/data.py -------------------------------------------------------------------------------- /mtcnn/train/gen_landmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/train/gen_landmark.py -------------------------------------------------------------------------------- /mtcnn/train/gen_onet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/train/gen_onet_train.py -------------------------------------------------------------------------------- /mtcnn/train/gen_pnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/train/gen_pnet_train.py -------------------------------------------------------------------------------- /mtcnn/train/gen_rnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/train/gen_rnet_train.py -------------------------------------------------------------------------------- /mtcnn/train/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/train/train_net.py -------------------------------------------------------------------------------- /mtcnn/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from . import draw -------------------------------------------------------------------------------- /mtcnn/utils/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/align_trans.py -------------------------------------------------------------------------------- /mtcnn/utils/cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/cp2tform.py -------------------------------------------------------------------------------- /mtcnn/utils/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/draw.py -------------------------------------------------------------------------------- /mtcnn/utils/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/functional.py -------------------------------------------------------------------------------- /mtcnn/utils/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mtcnn/utils/nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/cpu_nms.c -------------------------------------------------------------------------------- /mtcnn/utils/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /mtcnn/utils/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /mtcnn/utils/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /mtcnn/utils/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /mtcnn/utils/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/nms_kernel.cu -------------------------------------------------------------------------------- /mtcnn/utils/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/mtcnn/utils/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /output/caffe_models/det1.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det1.caffemodel -------------------------------------------------------------------------------- /output/caffe_models/det1.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det1.prototxt -------------------------------------------------------------------------------- /output/caffe_models/det2.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det2.caffemodel -------------------------------------------------------------------------------- /output/caffe_models/det2.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det2.prototxt -------------------------------------------------------------------------------- /output/caffe_models/det3.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det3.caffemodel -------------------------------------------------------------------------------- /output/caffe_models/det3.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det3.prototxt -------------------------------------------------------------------------------- /output/caffe_models/det4.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det4.caffemodel -------------------------------------------------------------------------------- /output/caffe_models/det4.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/output/caffe_models/det4.prototxt -------------------------------------------------------------------------------- /scripts/convert_caffe_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/convert_caffe_model.py -------------------------------------------------------------------------------- /scripts/detect_on_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/detect_on_video.py -------------------------------------------------------------------------------- /scripts/gen_onet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/gen_onet_train.py -------------------------------------------------------------------------------- /scripts/gen_pnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/gen_pnet_train.py -------------------------------------------------------------------------------- /scripts/gen_rnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/gen_rnet_train.py -------------------------------------------------------------------------------- /scripts/track_on_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/track_on_video.py -------------------------------------------------------------------------------- /scripts/train_onet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/train_onet.py -------------------------------------------------------------------------------- /scripts/train_pnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/train_pnet.py -------------------------------------------------------------------------------- /scripts/train_rnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/scripts/train_rnet.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/asset/images/audrey.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/asset/images/audrey.jpg -------------------------------------------------------------------------------- /tests/asset/images/bksomels.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/asset/images/bksomels.jpg -------------------------------------------------------------------------------- /tests/asset/images/gpripe.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/asset/images/gpripe.jpg -------------------------------------------------------------------------------- /tests/asset/images/office5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/asset/images/office5.jpg -------------------------------------------------------------------------------- /tests/asset/images/roate.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/asset/images/roate.jpg -------------------------------------------------------------------------------- /tests/asset/video/track.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/asset/video/track.mp4 -------------------------------------------------------------------------------- /tests/test_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_align.py -------------------------------------------------------------------------------- /tests/test_batch_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_batch_detection.py -------------------------------------------------------------------------------- /tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_data.py -------------------------------------------------------------------------------- /tests/test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_datasets.py -------------------------------------------------------------------------------- /tests/test_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_detection.py -------------------------------------------------------------------------------- /tests/test_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_functional.py -------------------------------------------------------------------------------- /tests/test_gptd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_gptd.py -------------------------------------------------------------------------------- /tests/test_landmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_landmarks.py -------------------------------------------------------------------------------- /tests/test_net_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_net_jit.py -------------------------------------------------------------------------------- /tests/test_net_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_net_pytorch.py -------------------------------------------------------------------------------- /tests/test_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_tracker.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tests/test_train.py -------------------------------------------------------------------------------- /tutorial/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tutorial/detect_step_by_step.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tutorial/detect_step_by_step.ipynb -------------------------------------------------------------------------------- /tutorial/face_align.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BrightXiaoHan/FaceDetector/HEAD/tutorial/face_align.ipynb --------------------------------------------------------------------------------