目录
- 0 前言
- 1 搭建环境与项目
- 2 人脸预测与结果展示
0 前言
这一次要复现的是人脸识别中的 facenet-pytorch
参考了:
Pytorch 搭建自己的Facenet人脸识别网络(Bubbliiiing 深度学习 教程)
https://gitee.com/xiaozhao123666/facenet-pytorch
https://github.com/bubbliiiing/facenet-pytorch
b站操作视频:
https://www.bilibili.com/video/BV1MG411u7Dv/
https://www.bilibili.com/video/BV1RG411M7rX/
本次采用windows的cpu进行复现
1 搭建环境与项目
环境:python=3.7 torch1.2 torchvision0.4.0 cu92
创建虚拟环境face
# 创建虚拟环境face
conda create --name face python=3.7 -y
#激活环境
conda activate face
下载人脸识别项目
git clone https://gitee.com/xiaozhao123666/facenet-pytorch.git
安装torch
#搭建项目所需pytorch环境
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
安装依赖
# 进入项目,安装依赖
cd facenet-pytorch
pip install -r requirements.txt
2 人脸预测与结果展示
在facenet-pytorch\img下存放如下图片
上图中,1_1、1_2、1_3为一个人,2_1是另一个人
该数据大家可以自己做,或者在下面的链接中下载(免费,我设置0下载券):
https://download.csdn.net/download/WhiffeYF/87209609
开始预测前,需要将facenet.py钟的cuda的值改为False(这样就是用cpu来运算,而不是用GPU),如下图。
开始预测
python predict.py
然后就会看到:
Input image_1 filename:
的字段,这就是让你输入第一张图片路径,共需要输入两张。
下面是检测结果