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

feat: updated face mesh lib class name

Browse Source
This commit is contained in:
dmaletskiy 2021-07-01 16:00:21 +03:00
parent de5fc2a532
commit c48c01ca75
3 changed files with 26 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/examples/desktop/face_mesh_dll/face_mesh_cpu.cpp
View File

@ -21,7 +21,7 @@ int main(int argc, char **argv) {
LOG(INFO) << "VideoCapture initialized.";
FaceMeshDetector *faceMeshDetector = FaceMeshDetector_Construct();
MPFaceMeshDetector *faceMeshDetector = FaceMeshDetector_Construct();
LOG(INFO) << "FaceMeshDetector constructed.";

30
mediapipe/examples/desktop/face_mesh_dll/face_mesh_lib.cpp
View File

@ -2,14 +2,14 @@
#include "face_mesh_lib.h"
FaceMeshDetector::FaceMeshDetector() {
MPFaceMeshDetector::MPFaceMeshDetector() {
const auto status = InitFaceMeshDetector();
if (!status.ok()) {
LOG(INFO) << "Failed constructing FaceMeshDetector.";
}
}
absl::Status FaceMeshDetector::InitFaceMeshDetector() {
absl::Status MPFaceMeshDetector::InitFaceMeshDetector() {
LOG(INFO) << "Get calculator graph config contents: " << graphConfig;
mediapipe::CalculatorGraphConfig config =
@ -37,8 +37,8 @@ absl::Status FaceMeshDetector::InitFaceMeshDetector() {
return absl::Status();
}
absl::Status FaceMeshDetector::ProcessFrameWithStatus(
cv::Mat &camera_frame,
absl::Status MPFaceMeshDetector::ProcessFrameWithStatus(
const cv::Mat &camera_frame,
std::unique_ptr<std::vector<std::vector<cv::Point2f>>>
&multi_face_landmarks) {
// Wrap Mat into an ImageFrame.
@ -115,7 +115,7 @@ absl::Status FaceMeshDetector::ProcessFrameWithStatus(
}
std::vector<std::vector<cv::Point2f>> *
FaceMeshDetector::ProcessFrame2D(cv::Mat &camera_frame) {
MPFaceMeshDetector::ProcessFrame2D(const cv::Mat &camera_frame) {
auto landmarks = std::make_unique<std::vector<std::vector<cv::Point2f>>>();
ProcessFrameWithStatus(camera_frame, landmarks);
@ -124,25 +124,27 @@ FaceMeshDetector::ProcessFrame2D(cv::Mat &camera_frame) {
}
extern "C" {
DLLEXPORT FaceMeshDetector *FaceMeshDetector_Construct() {
return new FaceMeshDetector();
DLLEXPORT MPFaceMeshDetector *FaceMeshDetector_Construct() {
return new MPFaceMeshDetector();
}
DLLEXPORT void FaceMeshDetector_Destruct(FaceMeshDetector *detector) {
DLLEXPORT void FaceMeshDetector_Destruct(MPFaceMeshDetector *detector) {
delete detector;
}
DLLEXPORT void *FaceMeshDetector_ProcessFrame2D(FaceMeshDetector *detector,
cv::Mat &camera_frame) {
DLLEXPORT void *
FaceMeshDetector_ProcessFrame2D(MPFaceMeshDetector *detector,
const cv::Mat &camera_frame) {
return reinterpret_cast<void *>(detector->ProcessFrame2D(camera_frame));
}
}
const char FaceMeshDetector::kInputStream[] = "input_video";
const char FaceMeshDetector::kOutputStream_landmarks[] = "multi_face_landmarks";
const char FaceMeshDetector::kOutputStream_faceCount[] = "face_count";
const char MPFaceMeshDetector::kInputStream[] = "input_video";
const char MPFaceMeshDetector::kOutputStream_landmarks[] =
"multi_face_landmarks";
const char MPFaceMeshDetector::kOutputStream_faceCount[] = "face_count";
const std::string FaceMeshDetector::graphConfig = R"pb(
const std::string MPFaceMeshDetector::graphConfig = R"pb(
# MediaPipe graph that performs face mesh with TensorFlow Lite on CPU.
# Input image. (ImageFrame)

18
mediapipe/examples/desktop/face_mesh_dll/face_mesh_lib.h
View File

@ -25,16 +25,16 @@
#include "mediapipe/framework/port/parse_text_proto.h"
#include "mediapipe/framework/port/status.h"
class FaceMeshDetector {
class MPFaceMeshDetector {
public:
FaceMeshDetector();
~FaceMeshDetector() = default;
std::vector<std::vector<cv::Point2f>> *ProcessFrame2D(cv::Mat &camera_frame);
MPFaceMeshDetector();
~MPFaceMeshDetector() = default;
std::vector<std::vector<cv::Point2f>> *ProcessFrame2D(const cv::Mat &camera_frame);
private:
absl::Status InitFaceMeshDetector();
absl::Status
ProcessFrameWithStatus(cv::Mat &camera_frame,
ProcessFrameWithStatus(const cv::Mat &camera_frame,
std::unique_ptr<std::vector<std::vector<cv::Point2f>>>
&multi_face_landmarks);
@ -54,12 +54,12 @@ private:
extern "C" {
#endif
DLLEXPORT FaceMeshDetector *FaceMeshDetector_Construct();
DLLEXPORT MPFaceMeshDetector *FaceMeshDetector_Construct();
DLLEXPORT void FaceMeshDetector_Destruct(FaceMeshDetector *detector);
DLLEXPORT void FaceMeshDetector_Destruct(MPFaceMeshDetector *detector);
DLLEXPORT void *FaceMeshDetector_ProcessFrame2D(FaceMeshDetector *detector,
cv::Mat &camera_frame);
DLLEXPORT void *FaceMeshDetector_ProcessFrame2D(MPFaceMeshDetector *detector,
const cv::Mat &camera_frame);
#ifdef __cplusplus
};