我正在尝试使用谷歌的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)
下面是输出。理想情况下,我不会在寮屋居民的右手边有任何圆点,脸上的圆点更少
谢谢
目前没有回答
相关问题 更多 >
编程相关推荐