从人脸地标点获取 ROI cv2 dlib

Get ROI from face landmark points cv2 dlib

我正在使用以下代码使用 dlib 在实时从网络摄像头捕获的帧上绘制面部标志点,现在我正在寻找的是获得边界符合面部的所有点的 ROI,代码如下如下:

import cv2
import dlib
import numpy
from imutils.video import FPS
from imutils.video import WebcamVideoStream
import imutils

PREDICTOR_PATH = "./shape_predictor_68_face_landmarks.dat"
predictor = dlib.shape_predictor(PREDICTOR_PATH)
cascade_path = 'cascade/haarcascade_frontalface_default.xml'
cascade = cv2.CascadeClassifier(cascade_path)

webcam = WebcamVideoStream(src=0).start()
fps = FPS().start()

while True:
    im = webcam.read()
    im = imutils.resize(im, width=400)

    faces = cascade.detectMultiScale(im, 1.3, 5)
    if len(faces) != 0:
        for (x, y, w, h) in faces.astype(long):
            rect = dlib.rectangle(x, y, x + w, y + h)
            #cv2.imwrite('face.png', rect)
            get_landmarks = numpy.matrix([[p.x, p.y] for p in predictor(im, rect).parts()])

        for idx, point in enumerate(get_landmarks):
            pos = (point[0, 0], point[0, 1])
            cv2.putText(im, str(idx), pos,
                        fontFace=cv2.FONT_HERSHEY_SCRIPT_SIMPLEX,
                        fontScale=0.4,
                        color=(0, 0, 255))
           # cv2.drawContours(im, [pos], -1, (0, 255, 0), 2)
#            hullIndex = cv2.convexHull(pos, returnPoints=False)
#            cv2.imwrite('face.png', hullIndex)
            cv2.circle(im, pos, 3, color=(0, 255, 255))
    cv2.imshow('Result', im)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
    fps.update()

cv2.destroyAllWindows()

我想将面部标志点转储到图像文件中。但无法理解,我如何从这些地标点提取 ROI,然后使用

转储到文件

cv2.imwrite('face.png', hullIndex)

,但这里我不希望使用

绘制矩形

faces = cascade.detectMultiScale(im, 1.3, 5)

,我只想要包围面部标志点的形状,上面代码的输出如下所示。

所以,只转储围绕面部边缘点的点。

从预测器获得 68 个面部标志后,您可以简单地遍历所有点并更新 minX, minY, maxX, maxY,迭代后将产生边界矩形,该矩形完全包围所有面部标志。

您也可以使用cv2.boundingRect(points),但在传递给此方法之前,您需要将所有点转换为 numpy 数组。

但我会向您推荐第一个解决方案,它很容易实现,因为

minX, minY = 10000000, 10000000
maxX, maxY = 0, 0
for point in get_landmarks:
    if point[0] < minX:
        minX = point[0]
    elif point[0] > maxX:
        maxX = point[0]
    elif point[1] < minY:
        minY = point[1]
    elif point[1] > maxY:
        maxY = point[1]

bounding_rect = [minX, minY, maxX - minX, maxY - minY]