MediaPipe姿势估计:是否可以仅标记某些地标点?

2024-05-12 20:27:43 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用谷歌的MediaPipe进行侧对侧姿势估计,我想知道是否可以只在身体右侧标记标志点?目前,我只连接了地标的右侧/左侧,但另一侧的标记点仍然显示出来

此外,我也不想显示除了耳朵和鼻子以外的面部标志点

这是我目前的代码

    def find_pose(self, frame, draw=True, box=False):
        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        self.results = self.pose.process(frame_rgb)
        self.landmarks = self.results.pose_landmarks
        pose_connections = self.landmark_connections.POSE_CONNECTIONS
        if self.landmarks:
            if draw:
                min_values, max_values = self.find_box_coordinates(frame)
                self.mpDraw.draw_landmarks(frame, self.landmarks, pose_connections)
                if box:
                    cv2.rectangle(frame, min_values, max_values, (0, 255, 0), 2)
        return frame

以及一些相关的变量声明:

import mediapipe as mp

self.mpDraw = mp.solutions.drawing_utils
self.mpPose = mp.solutions.pose
self.pose = self.mpPose.Pose(self.static_image_mode, self.model_complexity, self.smooth, self.detectionConf, self.trackingConf)

下面是输出。理想情况下,我不会在寮屋居民的右手边有任何圆点,脸上的圆点更少

sideonsquat

谢谢


Tags: 标记selfboxifmprgbfindconnections