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

Group landmarks points together with visibility in a single structure

Browse Source
This commit is contained in:
Maksym Walczak 2022-08-11 13:56:49 +02:00
parent 17d38c9c5a
commit 0c292576f9
2 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mediapipe/pose_tracking_dll/pose_tracking.cpp
View File

@ -153,12 +153,8 @@ bool PoseTracking::processFrame(const cv_wrapper::Mat& inputRGB8Bit) {
return mImplementation->processFrame(frame);
}
cv_wrapper::Point3f* PoseTracking::lastDetectedLandmarks() {
return mImplementation->lastDetectedLandmarks();
}
float* PoseTracking::lastLandmarksVisibility() {
return mImplementation->landmarksVisibility();
PoseTracking::PoseLandmarks PoseTracking::lastDetectedLandmarks() {
return {mImplementation->lastDetectedLandmarks(), mImplementation->landmarksVisibility()};
}
cv_wrapper::Mat PoseTracking::lastSegmentedFrame() {
@ -171,5 +167,4 @@ PoseTracking::~PoseTracking()
{
delete mImplementation;
}
} // namespace nimagna

12
mediapipe/pose_tracking_dll/pose_tracking.h
View File

@ -62,9 +62,16 @@ struct Mat {
};
} // namespace cv_wrapper
class DLLEXPORT PoseTracking {
public:
struct PoseLandmarks {
PoseLandmarks(cv_wrapper::Point3f* points, float* visibility) : points(points), visibility(visibility) {}
static constexpr size_t kLandmarksCount = 33u;
const cv_wrapper::Point3f* points;
const float* visibility;
};
enum LandmarkNames {
NOSE = 0,
LEFT_EYE_INNER,
@ -99,7 +106,7 @@ class DLLEXPORT PoseTracking {
RIGHT_HEEL,
LEFT_FOOT_INDEX,
RIGHT_FOOT_INDEX,
COUNT = kLandmarksCount
COUNT = PoseLandmarks::kLandmarksCount
};
PoseTracking(const char* calculatorGraphConfigFile);
@ -107,8 +114,7 @@ class DLLEXPORT PoseTracking {
bool processFrame(const cv_wrapper::Mat& inputRGB8Bit);
cv_wrapper::Mat lastSegmentedFrame();
cv_wrapper::Point3f* lastDetectedLandmarks();
float* lastLandmarksVisibility();
PoseTracking::PoseLandmarks lastDetectedLandmarks();
private:
PoseTrackingImpl* mImplementation;