Skip to content

Commit 1ff80f9

Browse files
MediaPipe Teamcopybara-github
MediaPipe Team
authored andcommitted
draw mouth to shoulder line after connection, to align with python viz code
PiperOrigin-RevId: 520935390
1 parent 4dcb9a2 commit 1ff80f9

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

mediapipe/util/pose_util.cc

+13-13
Original file line numberDiff line numberDiff line change
@@ -89,19 +89,6 @@ void DrawPose(const mediapipe::NormalizedLandmarkList& pose, int target_width,
8989

9090
constexpr int draw_line_width = 5;
9191
constexpr int draw_circle_radius = 7;
92-
const int lm = static_cast<int>(PoseLandmarkName::kMouthLeft);
93-
const int rm = static_cast<int>(PoseLandmarkName::kMouthRight);
94-
const int ls = static_cast<int>(PoseLandmarkName::kLeftShoulder);
95-
const int rs = static_cast<int>(PoseLandmarkName::kRightShoulder);
96-
if (visible_landmarks.find(lm) != visible_landmarks.end() &&
97-
visible_landmarks.find(rm) != visible_landmarks.end() &&
98-
visible_landmarks.find(ls) != visible_landmarks.end() &&
99-
visible_landmarks.find(rs) != visible_landmarks.end()) {
100-
cv::line(*image, (visible_landmarks[lm] + visible_landmarks[rm]) * 0.5f,
101-
(visible_landmarks[ls] + visible_landmarks[rs]) * 0.5f,
102-
cv::Scalar(255, 255, 255), draw_line_width);
103-
}
104-
10592
for (int j = 0; j < 35; ++j) {
10693
if (visible_landmarks.find(kJointConnection[j][0]) !=
10794
visible_landmarks.end() &&
@@ -115,6 +102,19 @@ void DrawPose(const mediapipe::NormalizedLandmarkList& pose, int target_width,
115102
}
116103
}
117104

105+
const int lm = static_cast<int>(PoseLandmarkName::kMouthLeft);
106+
const int rm = static_cast<int>(PoseLandmarkName::kMouthRight);
107+
const int ls = static_cast<int>(PoseLandmarkName::kLeftShoulder);
108+
const int rs = static_cast<int>(PoseLandmarkName::kRightShoulder);
109+
if (visible_landmarks.find(lm) != visible_landmarks.end() &&
110+
visible_landmarks.find(rm) != visible_landmarks.end() &&
111+
visible_landmarks.find(ls) != visible_landmarks.end() &&
112+
visible_landmarks.find(rs) != visible_landmarks.end()) {
113+
cv::line(*image, (visible_landmarks[lm] + visible_landmarks[rm]) * 0.5f,
114+
(visible_landmarks[ls] + visible_landmarks[rs]) * 0.5f,
115+
cv::Scalar(255, 255, 255), draw_line_width);
116+
}
117+
118118
for (const auto& landmark : visible_landmarks) {
119119
cv::circle(*image, landmark.second, draw_circle_radius,
120120
cv::Scalar(kJointColorMap[landmark.first][0],

0 commit comments

Comments
 (0)