当前位置: 首页 > article >正文

python学opencv|读取图像(七十四)人脸识别:EigenFaces算法

【1】引言

前序学习进程中,做的是检测,只是能检测出来由人脸、猫脸和行人,相关文章链接为:

python学opencv|读取图像(七十一)使用cv2.CascadeClassifier()函数+detectMultiScale()函数实现图像中的人脸检测-CSDN博客

python学opencv|读取图像(七十二)猫脸检测:使用cv2.CascadeClassifier()函数+detectMultiScale()函数实现图像中的猫脸检测-CSDN博客

python学opencv|读取图像(七十三)行人检测:使用cv2.CascadeClassifier()函数+detectMultiScale()函数实现图像中的行人检测-CSDN博客

更进一步,如果想在检测出来的人脸上识别出是谁的脸,就要用到人脸识别。

人脸识别的算法有三种,本次学习其中的EigenFaces算法。

【2】原理说明

人脸识别需要用到不同的算法,把算法理解成识别器。具体操作的时候,先挑选一种算法,然后使用这种算法进行训练,训练之后再去看识别效果。   

【3】官网教程

首先点击下方链接,直达人脸识别的说明页官网教程:

OpenCV: Face Recognition with OpenCV

在这里,非常清晰地说明了当前有三种人脸识别算法:

图1  人脸识别官网说明

官网页面内容非常丰富,但学习需要抓主要内容,因此只要关注Eigenfaces就可以。

当然,细心的朋友肯定也发现了,这个页面提供了一些照片集,大家可以自行下载。

继续下拉会看到,关于Eigenfaces给出了背后的算法证明:

图2  Eigenfaces算法证明

实际上,如果不是做研究,这个算法说空也不用太过深究,因为主要的精力要放在实现好的人脸识别上。

在这个页面之后,给出了C++的案例,就不继续研究这个页面了,请点击下方链接直达新的页面:

OpenCV: cv::face::EigenFaceRecognizer Class Reference

这个新页面给出了在python中如何通过Eigenfaces算法创建进人脸识别器:

图3  cv.face.EigenFaceRecognizer.create识别器

具体的,cv.face.EigenFaceRecognizer.create识别器有两个参数:

cv.face.EigenFaceRecognizer.create(    

num_components              #可选参数,PCA(主成分分析)中保留分量的个数

threshold  )                        #可选参数,人脸识别的阈值

创建识别器之后,需要对图像进行训练,所以还要用一个函数cv.face.FaceRecognizer.train()来创建训练器,点击下方链接直达:

OpenCV: cv::face::FaceRecognizer Class Reference

图4   用函数cv.face.FaceRecognizer.train()创建训练器

具体的,cv.face.FaceRecognizer.train()训练器有两个参数:

cv.face.FaceRecognizer.train( 

src                    #用来训练的样本图像,要求尺寸大小一致

labels )             #样本图像对应的标签

训练器训练完样本之后,可以使用cv.face.FaceRecognizer.predict()函数对待识别图像进行识别,点击下方链接,直达cv.face.FaceRecognizer.predict()函数的官网页面:

OpenCV: cv::face::FaceRecognizer Class Reference

具体的,cv.face.FaceRecognizer.predict()只有一个参数:

cv.face.FaceRecognizer.predict(

src            #待识别图像,要求与训练样本尺寸一致

)

cv.face.FaceRecognizer.predict(src)会输出两个结果:

label:待测图像和样本图像最佳匹配时,对应的样本图像标签值

confidence:匹配置信度,小于5000认为匹配效果好,0表示完全一样。

【4】代码测试

此处进行代码测试:

首先是引入必要的模块,定义两个空列表,分别用来存储图像和图像对应的标签:

import cv2 as cv #引入cv2模块
import numpy as np #引入numpy模块

photos = []  #定义photo空列表,用来存储图像
labels = []  #定义lables空列表,用来存储图像对应的标签

然后使用了3张图像作为样本来训练,这里要给出样本的存储地址:

# 定义图片路径列表
image_paths = [
    r"D:\python\pythonworkspace\pythonProject3\zyz\01.jpg",
    r"D:\python\pythonworkspace\pythonProject3\zyz\02.jpg",
    r"D:\python\pythonworkspace\pythonProject3\zyz\03.jpg"
]

为了把图像统一尺寸,需要先给定相关尺寸的约束条件:

#定义图像尺寸信息
width=600  #定义一个初始宽度量
height=800  #定义一个初始高度量
target_size = (width,height) #定义一个元组,存储宽度量和高度量

# 读取其余图片并调整尺寸
for path in image_paths[0:]:
    img = cv.imread(path, 0)
    if img is not None:
        # 调整图像尺寸为与第一张图片相同
        img = cv.resize(img, target_size)
        photos.append(img)
        labels.append(0)
    else:
        print(f"无法读取图片: {path}")

图像尺寸统一完成之后,需要给这一组图像加一个标签:

#定义标签
names = {"0": "su"}

然后创建人脸识别器:

# 创建人脸识别器
recognizer = cv.face.EigenFaceRecognizer.create()

然后训练人脸识别器:

# 训练人脸识别器
if photos and labels:
    recognizer.train(photos, np.array(labels)) #训练器读入人脸样本和对应标签

之后就是把待测图像拿进来进行测试。

先读取图像:

# 读取测试图片
    test_image_path = r"D:\python\pythonworkspace\pythonProject3\zyz\09.jpg"
    test_img = cv.imread(test_image_path, 0)

然后把图像尺寸修改成和样本统一的大小,之后就可以进行对比识别:

    if test_img is not None:
        # 调整测试图像尺寸与第一张图片一致
        test_img = cv.resize(test_img, target_size)
        label, confidence = recognizer.predict(test_img)

对于识别效果,需要按照匹配度再次处理:

# 设定识别阈值
        threshold = 5000  # 可根据实际情况调整
        if confidence > threshold:
            # 将灰度图转换为彩色图以便添加彩色文字
            test_img_color = cv.cvtColor(test_img, cv.COLOR_GRAY2BGR)
            # 添加未识别标记
            cv.putText(test_img_color, "Unrecognized", (10, 30), cv.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
            output_img = test_img_color
        else:
            print(f'confidence={confidence}')
            print(names[str(label)])
            # 将灰度图转换为彩色图
            output_img = cv.cvtColor(test_img, cv.COLOR_GRAY2BGR)

最后是输出图像:

        # 输出图像
        cv.imshow('Result Image', output_img)
        cv.waitKey(0)
        cv.destroyAllWindows()
    else:
        print(f"无法读取测试图片: {test_image_path}")

为了更好地理解,直接给出完整代码:

import cv2 as cv #引入cv2模块
import numpy as np #引入numpy模块

photos = []  #定义photo空列表,用来存储图像
labels = []  #定义lables空列表,用来存储图像对应的标签

# 定义图片路径列表
image_paths = [
    r"D:\python\pythonworkspace\pythonProject3\zyz\01.jpg",
    r"D:\python\pythonworkspace\pythonProject3\zyz\02.jpg",
    r"D:\python\pythonworkspace\pythonProject3\zyz\03.jpg"
]

#定义图像尺寸信息
width=600  #定义一个初始宽度量
height=800  #定义一个初始高度量
target_size = (width,height) #定义一个元组,存储宽度量和高度量

# 读取其余图片并调整尺寸
for path in image_paths[0:]:
    img = cv.imread(path, 0)
    if img is not None:
        # 调整图像尺寸为与第一张图片相同
        img = cv.resize(img, target_size)
        photos.append(img)
        labels.append(0)
    else:
        print(f"无法读取图片: {path}")
#定义标签
names = {"0": "su"}

# 创建人脸识别器
recognizer = cv.face.EigenFaceRecognizer.create()

# 训练人脸识别器
if photos and labels:
    recognizer.train(photos, np.array(labels)) #训练器读入人脸样本和对应标签
    # 读取测试图片
    test_image_path = r"D:\python\pythonworkspace\pythonProject3\zyz\09.jpg"
    test_img = cv.imread(test_image_path, 0)
    if test_img is not None:
        # 调整测试图像尺寸与第一张图片一致
        test_img = cv.resize(test_img, target_size)
        label, confidence = recognizer.predict(test_img)

        # 设定识别阈值
        threshold = 5000  # 可根据实际情况调整
        if confidence > threshold:
            # 将灰度图转换为彩色图以便添加彩色文字
            test_img_color = cv.cvtColor(test_img, cv.COLOR_GRAY2BGR)
            # 添加未识别标记
            cv.putText(test_img_color, "Unrecognized", (10, 30), cv.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
            output_img = test_img_color
        else:
            print(f'confidence={confidence}')
            print(names[str(label)])
            # 将灰度图转换为彩色图
            output_img = cv.cvtColor(test_img, cv.COLOR_GRAY2BGR)

        # 输出图像
        cv.imshow('Result Image', output_img)
        cv.waitKey(0)
        cv.destroyAllWindows()
    else:
        print(f"无法读取测试图片: {test_image_path}")

代码运行使用的训练样本均为女明星:

图5  训练样本

使用豆包AI生成的篮球明星进行识别,发现很好辨认出不是一个人:

图6 识别效果

【5】总结

掌握了python+opencv实现通过EigenFaces算法实现人脸识别的技巧。


http://www.kler.cn/a/557079.html

相关文章:

  • vue中将el-table导出为excel文件
  • 使用LangChain构建第一个ReAct Agent
  • C/C++ | 每日一练 (2)
  • 大模型掀起AI新浪潮,传感器成为智能升级关键
  • ABB机器人与西门子PLC、工业相机联合实现无序抓取的完整调试方案
  • ARM Linux下FFmpeg+Nginx+RTMP 视频监控
  • springboot+dubbo+zookeeper的注册服务和调用实践
  • 华为最新OD机试真题-找单词-Python-OD统一考试(E卷)
  • 【未完成】BUU20 [HCTF 2018]admin1
  • 代码随想录算法训练营day40(补0208)
  • 在Ubutu18.04下搭建nfs服务器
  • 用DeepSeek来帮助学习three.js加载3D太极模形
  • 阅读论文笔记《Translating Embeddings for Modeling Multi-relational Data》
  • Javascript使用Sodium库实现 aead_xchacha20poly1305_ietf加密解密,以及与后端的密文交互
  • HarmonyOS NEXT全栈开发深度指南(API 12+)
  • 计算机网络之物理层——基于《计算机网络》谢希仁第八版
  • oracle主库添加数据文件后备库无法按convert转换数据文件名ORA-01119,ORA-17502,ORA-15001
  • 如何查看java的字节码文件?javap?能用IDEA吗?
  • 初级银行从业考试真题
  • uniapp setInterval() 循环调用clearInterval()清除调用