facenet快速复现 实现 facenet-pytorch 人脸识别 windows上 使用cpu实现 人脸对比

Posted 2023-02-09 CV-杨帆

目录

• 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
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92  -f https://download.pytorch.org/whl/torch_stable.html

安装依赖

# 进入项目，安装依赖
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:

的字段，这就是让你输入第一张图片路径，共需要输入两张。
下面是检测结果