cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

apply affine transform before drawing, in order to keep constant line width regardless of face cropping.

Browse Source 
PiperOrigin-RevId: 555173659
This commit is contained in:
MediaPipe Team 2023-08-09 08:41:04 -07:00 committed by Copybara-Service
parent 00e0314040
commit a9c7e22ca4
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mediapipe/util/pose_util.cc
View File

@ -192,15 +192,20 @@ void DrawPose(const mediapipe::NormalizedLandmarkList& pose, bool flip_y,
} }
} }
void DrawFace(const mediapipe::NormalizedLandmarkList& face, bool flip_y, void DrawFace(const mediapipe::NormalizedLandmarkList& face,
bool draw_nose, int color_style, bool reverse_color, const std::pair<int, int>& image_size, const cv::Mat& affine,
bool flip_y, bool draw_nose, int color_style, bool reverse_color,
int draw_line_width, cv::Mat* image) { int draw_line_width, cv::Mat* image) {
const int target_width = image->cols;
const int target_height = image->rows;
std::vector<cv::Point> landmarks; std::vector<cv::Point> landmarks;
for (const auto& lm : face.landmark()) { for (const auto& lm : face.landmark()) {
landmarks.emplace_back(lm.x() * target_width, float ori_x = lm.x() * image_size.first;
(flip_y ? 1.0f - lm.y() : lm.y()) * target_height); float ori_y = (flip_y ? 1.0f - lm.y() : lm.y()) * image_size.second;
landmarks.emplace_back(
affine.at<float>(0, 0) * ori_x + affine.at<float>(0, 1) * ori_y +
affine.at<float>(0, 2),
affine.at<float>(1, 0) * ori_x + affine.at<float>(1, 1) * ori_y +
affine.at<float>(1, 2));
} }
cv::Scalar kFaceOvalColor; cv::Scalar kFaceOvalColor;

5
mediapipe/util/pose_util.h
View File

@ -23,8 +23,9 @@ namespace mediapipe {
void DrawPose(const mediapipe::NormalizedLandmarkList& pose, bool flip_y, void DrawPose(const mediapipe::NormalizedLandmarkList& pose, bool flip_y,
cv::Mat* image); cv::Mat* image);
void DrawFace(const mediapipe::NormalizedLandmarkList& face, bool flip_y, void DrawFace(const mediapipe::NormalizedLandmarkList& face,
bool draw_nose, int color_style, bool reverse_color, const std::pair<int, int>& image_size, const cv::Mat& affine,
bool flip_y, bool draw_nose, int color_style, bool reverse_color,
int draw_line_width, cv::Mat* image); int draw_line_width, cv::Mat* image);
} // namespace mediapipe } // namespace mediapipe