人脸关键点检测模型-MTCNN

P粉084495128

发布时间：2025-07-18 14:32:22

215人浏览过

来源于php中文网

原创

本文介绍了基于PaddlePaddle实现的MTCNN多任务卷积神经网络项目，该模型含P-Net、R-Net、O-Net三层结构，可检测人脸区域及关键点。说明环境要求、文件功能，详述数据集下载与放置，训练需分三步依次训练PNet、RNet、ONet模型，每步依赖上一步结果，还提及预测方法及参考资料。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

人脸关键点检测模型-mtcnn - php中文网

啥也别说，先来试试效果吧。

In [2]

!python infer_path.py --image_path=dataset/test.jpgfrom PIL import Image
display(Image.open('result.jpg'))

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/setuptools/depends.py:2: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
  import imp
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/backward.py:1640: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated, and in 3.8 it will stop working
  return list(x) if isinstance(x, collections.Sequence) else [x]
W0408 11:14:58.244475   362 device_context.cc:362] Please NOTE: device: 0, GPU Compute Capability: 7.0, Driver API Version: 10.1, Runtime API Version: 10.1
W0408 11:14:58.249330   362 device_context.cc:372] device: 0, cuDNN Version: 7.6.
: cannot connect to X server

前言

MTCNN，Multi-task convolutional neural network（多任务卷积神经网络），将人脸区域检测与人脸关键点检测放在了一起，总体可分为P-Net、R-Net、和O-Net三层网络结构。它是2016年中国科学院深圳研究院提出的用于人脸检测任务的多任务神经网络模型，该模型主要采用了三个级联的网络，采用候选框加分类器的思想，进行快速高效的人脸检测。这三个级联的网络分别是快速生成候选窗口的P-Net、进行高精度候选窗口过滤选择的R-Net和生成最终边界框与人脸关键点的O-Net。和很多处理图像问题的卷积神经网络模型，该模型也用到了图像金字塔、边框回归、非最大值抑制等技术。

环境

PaddlePaddle 2.0.1
Python 3.7

文件介绍

models/Loss.py MTCNN所使用的损失函数，包括分类损失函数、人脸框损失函数、关键点损失函数
models/PNet.py PNet网络结构
models/RNet.py RNet网络结构
models/ONet.py ONet网络结构
utils/data_format_converter.py 把大量的图片合并成一个文件
utils/data.py 训练数据读取器
utils/utils.py 各种工具函数
train_PNet/generate_PNet_data.py 生成PNet训练的数据
train_PNet/train_PNet.py 训练PNet网络模型
train_RNet/generate_RNet_data.py 生成RNet训练的数据
train_RNet/train_RNet.py 训练RNet网络模型
train_ONet/generate_ONet_data.py 生成ONet训练的数据
train_ONet/train_ONet.py 训练ONet网络模型
infer_path.py 使用路径预测图像，检测图片上人脸的位置和关键的位置，并显示
infer_camera.py 预测图像程序，检测图片上人脸的位置和关键的位置实时显示

数据集下载

WIDER Face 下载训练数据WIDER Face Training Images，解压的WIDER_train文件夹放置到dataset下。并下载 Face annotations ，解压把里面的 wider_face_train_bbx_gt.txt 文件放在dataset目录下，
Deep Convolutional Network Cascade for Facial Point Detection 。下载 Training set 并解压，将里面的 lfw_5590 和 net_7876 文件夹放置到dataset下
解压数据集之后，dataset目录下应该有文件夹lfw_5590，net_7876，WIDER_train，有标注文件testImageList.txt，trainImageList.txt，wider_face_train.txt，wider_face_train_bbx_gt.txt

训练模型

训练模型一共分为三步，分别是训练PNet模型、训练RNet模型、训练ONet模型，每一步训练都依赖上一步的结果。

第一步训练PNet模型

PNet全称为Proposal Network，其基本的构造是一个全卷积网络，P-Net是一个人脸区域的区域建议网络，该网络的将特征输入结果三个卷积层之后，通过一个人脸分类器判断该区域是否是人脸，同时使用边框回归。

人脸关键点检测模型-MTCNN - php中文网

请在终端执行以下命令：

Transor

专业的AI翻译工具，支持网页、字幕、PDF、图片实时翻译

下载

cd train_PNet 切换到train_PNet文件夹
python3 generate_PNet_data.py 首先需要生成PNet模型训练所需要的图像数据
python3 train_PNet.py 开始训练PNet模型

第二步训练RNet模型

全称为Refine Network，其基本的构造是一个卷积神经网络，相对于第一层的P-Net来说，增加了一个全连接层，因此对于输入数据的筛选会更加严格。在图片经过P-Net后，会留下许多预测窗口，我们将所有的预测窗口送入R-Net，这个网络会滤除大量效果比较差的候选框，最后对选定的候选框进行Bounding-Box Regression和NMS进一步优化预测结果。

人脸关键点检测模型-MTCNN - php中文网

请在终端执行以下命令：