Skip to content

xiezheng-cs/mtcnn_pytorch

Repository files navigation

人脸检测


实验目的

  1. 理解和掌握基于神经网络的人脸检测方法的理论基础知识。
  2. 理解MTCNN人脸检测的基本流程,并加以实践。

实验环境

anaconda3 pytorch 0.4.1 torchvision opencv-python tensorflow, tensorboard 等。

实验步骤

一、获取代码

实验完整代码mtcnn_pytorch,可直接下载或是通过git clone命令下载。

git clone https://github.com/xiezheng-cs/mtcnn_pytorch.git

二、实验环境安装

  1. 确保本机或是服务器已安装好anaconda3环境;
  2. pip或conda安装pytorch 0.4.1 和 torchvision环境;
  3. pip或conda安装opencv-python环境;
  4. pip或conda按照tensorflow和tensorboard.
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl     # Windows
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl  # Linux
pip install torchvision
pip install opencv-python

三、简单测试给定模型

直接使用我们训练好的网络模型在给定的测试数据集(位于mtcnn_pytorch/data/test_images/目录下,共64张测试图片),运行以下命令,即可在mtcnn_pytorch/data/you_result/目录下查看检测结果。

cd mtcnn_pytorch/
python test_image.py