opencv 65 face mesh mediapipe

#face_mesh.py

import os

import cv2

import mediapipe as mp

cap = cv2.VideoCapture("assets/face mesh.mp4")

# Define the codec and create VideoWriter object

fps = 25.175

frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))

frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

size = (int(frame_width), int(frame_height))

fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')

path = 'C:/Users/zchen/PycharmProjects/opencv/googleNet/record'

out = cv2.VideoWriter()

success = out.open(os.path.join(path, "face mesh mediapipe.mov"), fourcc, fps, size, True)

mpFaceMesh = mp.solutions.face_mesh

faceMesh = mpFaceMesh.FaceMesh(max_num_faces=3)

mpDraw = mp.solutions.drawing_utils

drawSpec = mpDraw.DrawingSpec(thickness=1, circle_radius=1, color=(0, 0, 255))

while True:

    ret, frame = cap.read()

    if ret == False: break

    frame_RGB = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

    results = faceMesh.process(frame_RGB)

    if results.multi_face_landmarks:

        for faceLms in results.multi_face_landmarks:

            for id, lm in enumerate(faceLms.landmark):

                #print(id, lm)

                h, w, c = frame.shape

                cx, cy = int(lm.x * w), int(lm.y * h)

                #print(id, cx, cy)

            mpDraw.draw_landmarks(frame, faceLms, mpFaceMesh.FACE_CONNECTIONS,

                                  drawSpec, drawSpec)

    cv2.imshow("frame", frame)

    out.write(frame)

    if cv2.waitKey(1) == ord('q'):

        break

    if cv2.waitKey(1) == ord('p'):

        cv2.waitKey(-1)  # wait until any key is pressed

cap.release()

cv2.destroyAllWindows()

reference:

https://www.youtube.com/watch?v=V9bzew8A1tc