当前位置:   article > 正文

少走弯路:OpenCV、insightface 等多方案人脸推理和识别_facedetector = cv2.facedetectoryn.create

facedetector = cv2.facedetectoryn.create

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

在下图中查找脸部

第一种方案:

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

  1. import numpy as np
  2. import cv2
  3. img=cv2.imread("00000523.jpg")
  4. # img=cv2.resize(img, new_shape)
  5. faceDetector=cv2.FaceDetectorYN.create("opencv_zoo/models/face_detection_yunet/face_detection_yunet_2023mar.onnx","",(img.shape[1],img.shape[0]))
  6. faces = faceDetector.detect(img)
  7. f = faces[1]
  8. for ps in f:
  9. x=int(ps[0])
  10. y=int(ps[1])
  11. w=int(ps[2])
  12. h=int(ps[3])
  13. s=ps[14]
  14. # print(x,y,w,h,s)
  15. cv2.rectangle(img, (x, y), (x+w, y+h), (0,0,255), 2)
  16. cv2.namedWindow("d",0);
  17. cv2.resizeWindow("d", int(img.shape[1]/2), int(img.shape[0]/2));
  18. cv2.imshow("d",img)
  19. cv2.waitKey()
  20. # cv2.destoryAllWindows()
  21. cv2.destroyAllWindows()

结果如下:

第二种方案:

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

  1. import numpy as np
  2. import cv2
  3. img=cv2.imread("00000523.jpg")
  4. cas_default = cv2.CascadeClassifier("D:\\opencv-4.52\\data\\haarcascades\\haarcascade_frontalface_default.xml")
  5. faces_default = cas_default.detectMultiScale(img)
  6. for (x,y,w,h) in faces_default:
  7. print(x,y,w,h)
  8. cv2.rectangle(img, (x, y), (x+w, y+h), (0,255,255), 2)
  9. cv2.namedWindow("d",0);
  10. cv2.resizeWindow("d", int(img.shape[1]/2), int(img.shape[0]/2));
  11. cv2.imshow("d",img)
  12. cv2.waitKey()
  13. # cv2.destoryAllWindows()
  14. cv2.destroyAllWindows()

结果如下:

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

然后是insightface 的方案:

  1. import insightface
  2. import cv2
  3. import numpy as np
  4. model = insightface.app.FaceAnalysis()
  5. model.prepare(ctx_id=0, det_thresh=0.45)
  6. face_img = cv2.imread('00000523.jpg')
  7. rgb_small_frame = face_img[:, :, ::-1]
  8. faces = model.get(rgb_small_frame)
  9. for r in faces:
  10. box=r.bbox.astype(int)
  11. color=(0, 0, 255)
  12. cv2.rectangle(face_img, (box[0], box[1]), (box[2], box[3]), color, 2)
  13. cv2.namedWindow("d",0);
  14. cv2.resizeWindow("d", int(face_img.shape[1]/2), int(face_img.shape[0]/2));
  15. cv2.imshow("d",face_img)
  16. cv2.waitKey()
  17. cv2.destroyAllWindows()

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

这是原图:

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

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

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

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/116175
推荐阅读
相关标签
  

闽ICP备14008679号