项目作者: foamliu

项目描述 :
MTCNN 论文预测部分的 PyTorch 实现
高级语言: Python
项目地址: git://github.com/foamliu/MTCNN.git
创建时间: 2018-12-03T06:47:47Z
项目社区:https://github.com/foamliu/MTCNN

开源协议:Apache License 2.0

关键词:
face-alignment face-detection

下载


MTCNN

MTCNN 论文预测部分的 PyTorch 实现:
Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks.

模型

管道

级联框架管道包括三阶段多任务深度卷积网络:

  1. 首先,短平快的Proposal Network (P-Net) 生成候选窗口;
  2. 然后在下一阶段通过 Refinement Network (R-Net) 完善这些窗口;
  3. 最后阶段,Output Network (O-Net) 生成边界框和脸部标记点。

image

架构

P-Net, R-Net, and O-Net 的模型架构,其中 “MP” 表示最大池化,“Conv”表示卷积。

image

效果

image

用法

  1. $ python demo.py

鸣谢

基于项目: