From 0ed01b9172fddceda411b681ae0522ecd313a256 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laure=CE=B7t?= Date: Fri, 25 Nov 2022 11:20:04 +0100 Subject: [PATCH] chore: move code to main function --- src/main.py | 98 ++++++++++++++++++++++++++++++++--------------------- 1 file changed, 59 insertions(+), 39 deletions(-) diff --git a/src/main.py b/src/main.py index 3a5ee8e..388e623 100644 --- a/src/main.py +++ b/src/main.py @@ -1,48 +1,68 @@ +import logging + import cv2 import mediapipe as mp -mp_drawing = mp.solutions.drawing_utils # type: ignore -mp_drawing_styles = mp.solutions.drawing_styles # type: ignore -mp_holistic = mp.solutions.holistic # type: ignore -# open webcam -cap = cv2.VideoCapture(0) +def main() -> None: + """Main function.""" + mp_drawing = mp.solutions.drawing_utils # type: ignore + mp_drawing_styles = mp.solutions.drawing_styles # type: ignore + mp_holistic = mp.solutions.holistic # type: ignore -with mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5) as holistic: - while cap.isOpened(): - # read webcam - success, image = cap.read() - if not success: - print("Ignoring empty camera frame.") - # If loading a video, use 'break' instead of 'continue' - continue + # open webcam + cap = cv2.VideoCapture(0) - # to improve performance, optionally mark the image as not writeable to pass by reference - image.flags.writeable = False - image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) - results = holistic.process(image) + with mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5) as holistic: + while cap.isOpened(): + # read webcam + success, image = cap.read() + if not success: + print("Ignoring empty camera frame.") + # If loading a video, use 'break' instead of 'continue' + continue - # draw landmark annotation on the image - image.flags.writeable = True - image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) - mp_drawing.draw_landmarks( - image, - results.face_landmarks, - mp_holistic.FACEMESH_CONTOURS, - landmark_drawing_spec=None, - connection_drawing_spec=mp_drawing_styles.get_default_face_mesh_contours_style(), - ) - mp_drawing.draw_landmarks( - image, - results.pose_landmarks, - mp_holistic.POSE_CONNECTIONS, - landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(), - ) + # to improve performance, optionally mark the image as not writeable to pass by reference + image.flags.writeable = False + image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) + results = holistic.process(image) - # flip the image horizontally for a selfie-view display. - cv2.imshow("MediaPipe Holistic", cv2.flip(image, 1)) - if cv2.waitKey(5) & 0xFF == 27: - break + # log landmarks + logging.debug(results.face_landmarks) + logging.debug(results.pose_landmarks) -# close webcam -cap.release() + # draw landmark annotation on the image + image.flags.writeable = True + image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) + mp_drawing.draw_landmarks( + image, + results.face_landmarks, + mp_holistic.FACEMESH_CONTOURS, + landmark_drawing_spec=None, + connection_drawing_spec=mp_drawing_styles.get_default_face_mesh_contours_style(), + ) + mp_drawing.draw_landmarks( + image, + results.pose_landmarks, + mp_holistic.POSE_CONNECTIONS, + landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style(), + ) + + # flip the image horizontally for a selfie-view display. + cv2.imshow("MediaPipe Holistic", cv2.flip(image, 1)) + if cv2.waitKey(5) & 0xFF == 27: + break + + # close webcam + cap.release() + + +if __name__ == "__main__": + # setup logging format + logging.basicConfig( + level=logging.DEBUG, + format="%(asctime)s %(name)s %(levelname)-8s %(message)s", + datefmt="(%F %T)", + ) + + main()