pytorch快速上手-几行代码完成人脸识别验证

最新推荐文章于 2024-08-30 17:53:55 发布

活塞君

最新推荐文章于 2024-08-30 17:53:55 发布

阅读量704

点赞数

分类专栏：深度学习 pytorch 人脸识别文章标签： pytorch python

本文链接：https://blog.csdn.net/weixin_48174100/article/details/120152970

版权

这篇博客介绍了如何使用PyTorch快速实现人脸识别。首先讲解了安装facenet的步骤，接着阐述了如何构建验证数据库，包括利用mtcnn提取脸部图片并保存全连接输出。最后，详细说明了人脸识别过程，通过提取人脸特征并与数据库对比来确定身份。

摘要由CSDN通过智能技术生成

安装

本文注重快速落地，有不懂的另行百度
首先安装facenet

pip install facenet_pytorch

验证数据库构造

先用mtcnn提取脸部图片

import cv2
from facenet_pytorch import MTCNN,InceptionResnetV1
resnet=InceptionResnetV1(pretrained='vggface2').eval()
mtcnn = MTCNN(keep_all='true') #如果只图片只有一个人 这个true要去掉 然后要用unsqueeze加一层
##验证图片：
img_path1 = r'images\hongqing1.JPG'
img1 = cv2.imread(img_path1)
#中间省略
imgdi = {
   'hongqing':img1}