基于dlib的视频关键点检测

1 加入库

import cv2
import dlib
import numpy as np

capture = cv2.VideoCapture(0)

detector = dlib.get_frontal_face_detector()

predictor = dlib.shape_predictor(‘shape_predictor_68_face_landmarks.dat’)

if capture.isOpened() is False:
print(‘Camera error’)

def main(): #遗漏主函数，导致视频一直打不开

while True:
    ret, frame = capture.read()
    if ret:
        # frame = frame[:, :, ::-1]
        # frame = frame.astype(np.uint8)   调用视频的时候居然不需要RGB转变！奇怪了，图片居然又需要！！
        gray = cv2.flip(frame, 1)
        grays = cv2.cvtColor(gray, cv2.COLOR_BGR2GRAY)
   
    faces = detector(grays, 1)  

   
    for face in faces:    # (x, y, w, h)

      
        cv2.rectangle(gray.copy(), (face.left(), face.top()), (face.right(), face.bottom()), (0, 255, 0), 5)

       
        # shape = predictor(frame, face)
        shape = predictor(gray,