【实习】Facenet人脸识别

1.环境配置

使用 conda 安装：CUDA 12.1

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

在激活的环境中，使用 pip 安装剩余的依赖包：

pip install tensorboard scikit-learn scipy numpy matplotlib opencv-python tqdm Pillow h5py

2.代码运行

bubbliiiing/facenet-pytorch： 这是一个facenet-pytorch的库，可以用于训练自己的人脸识别模型。

项目结构 

将测试集放到lfw文件下，将训练好的模型放在model_data文件夹下。

运行predict.py开始运行

在测试集中选择一张图片。

之后开始对比这张图片与各个文件夹中图片的相似度。

 

选择相似度最小的那一张。得到判断结果。