少走弯路:OpenCV、insightface 等多方案人脸推理和识别

news/2024/7/23 11:04:09 标签: opencv, 人工智能, 计算机视觉, insightface

脑壳有包又花时间折腾了一下,其实之前也折腾过,主要是新看了一个方法

在下图中查找脸部

第一种方案:

使用了opencv 的cv2.FaceDetectorYN. ,完整代码如下:

import numpy as np
import cv2

img=cv2.imread("00000523.jpg")
# img=cv2.resize(img, new_shape)
faceDetector=cv2.FaceDetectorYN.create("opencv_zoo/models/face_detection_yunet/face_detection_yunet_2023mar.onnx","",(img.shape[1],img.shape[0]))
faces = faceDetector.detect(img)
f = faces[1]
for ps in f:
    x=int(ps[0])
    y=int(ps[1])
    w=int(ps[2])
    h=int(ps[3])
    s=ps[14]
    # print(x,y,w,h,s)
    cv2.rectangle(img, (x, y), (x+w, y+h), (0,0,255), 2)
cv2.namedWindow("d",0);
cv2.resizeWindow("d", int(img.shape[1]/2), int(img.shape[0]/2));
cv2.imshow("d",img)
cv2.waitKey()
# cv2.destoryAllWindows()
cv2.destroyAllWindows()

结果如下:

第二种方案:

使用了 cv2.CascadeClassifier("D:\\opencv-4.52\\data\\haarcascades\\haarcascade_frontalface_default.xml")

import numpy as np
import cv2

img=cv2.imread("00000523.jpg")
cas_default = cv2.CascadeClassifier("D:\\opencv-4.52\\data\\haarcascades\\haarcascade_frontalface_default.xml")
faces_default = cas_default.detectMultiScale(img)
for (x,y,w,h) in faces_default:
    print(x,y,w,h)
    cv2.rectangle(img, (x, y), (x+w, y+h), (0,255,255), 2)
cv2.namedWindow("d",0);
cv2.resizeWindow("d", int(img.shape[1]/2), int(img.shape[0]/2));
cv2.imshow("d",img)
cv2.waitKey()
# cv2.destoryAllWindows()
cv2.destroyAllWindows()

结果如下:

以上两种都是opencv的自带方案,要么找脸不全,要么错误的找脸

然后是insightface 的方案:

import insightface
import cv2
import numpy as np

model = insightface.app.FaceAnalysis()
model.prepare(ctx_id=0, det_thresh=0.45)

face_img = cv2.imread('00000523.jpg')
rgb_small_frame = face_img[:, :, ::-1]
faces = model.get(rgb_small_frame)

for r in faces:
    box=r.bbox.astype(int)
    color=(0, 0, 255)
    cv2.rectangle(face_img, (box[0], box[1]), (box[2], box[3]), color, 2)
cv2.namedWindow("d",0);
cv2.resizeWindow("d", int(face_img.shape[1]/2), int(face_img.shape[0]/2));
cv2.imshow("d",face_img)
cv2.waitKey()
cv2.destroyAllWindows()

汇总的就是: insightface  脸部的查找最好,稍微有点慢,不知道是不是因为推理了年龄和性别。用insightface  推理出来的脸部数据,来识别具体人物的准确度也很高。

这是原图:

对于推理出来的人脸进行分别标记,对应到一个数据集:

参照标记的脸部数据,对其他照片进行推理,以下是结果,就不贴代码了:

以下结果展示了不同角度下,都给了一个准确的结果。不要介意对于性别和年龄的推断,可能同亚洲和欧洲人种数据集有关系。

带了墨镜居然都有一个准确结果


http://www.niftyadmin.cn/n/5236779.html

相关文章

leetcode_1423 可获得的最大点数

1. 题意 给定一个数组,每次只能从头和尾进行选择。 选择k次当前头或者尾,问能取到的最大值。 可获得的最大点数 2. 题解 主要难点是意识到这是一个滑动窗口问题。 2.1 滑动窗口 令数组长度为 s z sz sz 令 s _ w ( p o s , k ) s\_w(pos, k) s_w(po…

scikit-learn线性回归法进行利润预测

大家好,生成式人工智能无疑是一个改变游戏规则的技术,但对于大多数商业问题来说,回归和分类等传统的机器学习模型仍然是首选。 私募股权或风险投资这样的投资者利用机器学习,首先必须了解关注的数据以及它是如何被使用的。投资公…

kernel | 不想老是编译内核?sysfs和debugfs了解一下

编译内核是一件让大家都抗拒的事情,因为编译一次内核需要的时间成本比较漫长,而且如果每次代码的微小改动或者想要额外调用某一个函数执行某一个动作就要不断的编译内核的话,就相当于CPU大量的时间都用在了idle,开发效率将会是相当…

深度学习——第3章 Python程序设计语言(3.2 Python程序流程控制)

3.2 Python程序流程控制 目录 1.布尔数据类型及相关运算 2.顺序结构 3.选择(分支)结构 4.循环结构 无论是在机器学习还是深度学习中,Python已经成为主导性的编程语言。而且,现在许多主流的深度学习框架,例如PyTorc…

【23-24 秋学期】NNDL 作业10 BPTT

习题6-1P 推导RNN反向传播算法BPTT. 习题6-2 推导公式(6.40)和公式(6.41)中的梯度. 习题6-3 当使用公式(6.50)作为循环神经网络的状态更新公式时, 分析其可能存在梯度爆炸的原因并给出解决方法. 习题6-2P 设计简单RNN模型,分别…

【腾讯云HAI域探密】- AIGC应用助力企业降本增效之路

一、前言: 近年来,随着深度学习、大数据、人工智能、AI等技术领域的不断发展,机器学习是目前最火热的人工智能分支之一,是使用大量数据训练计算机程序,以实现智能决策、语音识别、图像处理等任务。 作者也是经过了以…

第四章 React之Typescript

一、专栏介绍 欢迎加入本专栏!我将带领您从零开始快速掌握React,从搭建项目到深入理解React项目。后续还会将主流的Umi Max作为前端应用框架,并借助Ant Design Pro来设计用户界面。在这个专栏中,我将为您揭示开发过程中常见功能的…

【数据挖掘】国科大刘莹老师数据挖掘课程作业 —— 第一次作业

homewrok 1 1. 假定数据仓库中包含 4 个维:date, product, vendor, location;和两个度量:sales_volume 和 sales_cost。 (a) 画出该数据仓库的星形模式 图 1 星形模式图 (b) 由基本方体 [date, product, vendor, location] 开始&#x…