meidapipe输出归一坐标转换为相对坐标
时间: 2024-02-17 10:04:23 浏览: 498
如果你使用的是Mediapipe库中的姿态估计模块,可以通过以下方式将Mediapipe输出的归一化坐标转换为相对坐标:
1. 获取图像的宽度和高度:首先需要从输入图像中获取宽度和高度,可以使用OpenCV库中的函数`cv::Mat::cols`和`cv::Mat::rows`获取。
2. 将归一化坐标转换为像素坐标:使用图像的宽度和高度将Mediapipe输出的归一化坐标转换为像素坐标。具体来说,Mediapipe输出的归一化坐标x和y的值分别乘以图像的宽度和高度,即可得到对应的像素坐标。
3. 计算相对坐标:计算相对坐标需要知道参考点的坐标。参考点可以是图像的中心点、左上角点等。例如,如果将左上角点作为参考点,那么相对坐标就是归一化坐标与左上角点像素坐标之差。具体来说,相对坐标x的值等于归一化坐标x乘以图像宽度减去左上角点的像素坐标x值,相对坐标y的值等于归一化坐标y乘以图像高度减去左上角点的像素坐标y值。
下面是一个示例代码,展示了如何将Mediapipe输出的归一化坐标转换为相对坐标:
```
#include <opencv2/opencv.hpp>
#include "mediapipe/framework/formats/landmark.pb.h" // mediapipe::NormalizedLandmarkList
// 将Mediapipe的归一化坐标转换为相对坐标
void ConvertNormalizedLandmarkListToPoint2f(const mediapipe::NormalizedLandmarkList& landmarks,
std::vector<cv::Point2f>* points) {
points->clear();
const int num_landmarks = landmarks.landmark_size();
points->reserve(num_landmarks);
for (int i = 0; i < num_landmarks; ++i) {
const auto& landmark = landmarks.landmark(i);
const float x = landmark.x();
const float y = landmark.y();
points->emplace_back(x, y);
}
}
// 计算相对坐标
void CalculateRelativeCoordinates(const std::vector<cv::Point2f>& landmarks,
const cv::Size& image_size,
std::vector<cv::Point2f>* relative_landmarks) {
relative_landmarks->clear();
relative_landmarks->reserve(landmarks.size());
const cv::Point2f origin(0.0f, 0.0f);
for (const auto& landmark : landmarks) {
const cv::Point2f point(static_cast<float>(landmark.x) * image_size.width,
static_cast<float>(landmark.y) * image_size.height);
relative_landmarks->emplace_back(point - origin);
}
}
int main() {
mediapipe::NormalizedLandmarkList landmarks;
// 获取Mediapipe输出的归一化坐标
// ...
std::vector<cv::Point2f> points;
ConvertNormalizedLandmarkListToPoint2f(landmarks, &points);
const cv::Mat image = cv::imread("image.jpg");
const cv::Size image_size(image.cols, image.rows);
std::vector<cv::Point2f> relative_landmarks;
CalculateRelativeCoordinates(points, image_size, &relative_landmarks);
// 使用相对坐标进行后续操作
// ...
return 0;
}
```
需要注意的是,以上代码中的`mediapipe::NormalizedLandmarkList`类型是Mediapipe库中的类型,在实际使用中需要根据自己的情况进行修改。另外,此处使用了OpenCV库中的`cv::Point2f`和`cv::Size`类型,需要包含OpenCV头文件。
阅读全文