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

Internal change

Browse Source 
PiperOrigin-RevId: 547258228
This commit is contained in:
MediaPipe Team 2023-07-11 12:03:30 -07:00 committed by Copybara-Service
parent bf6561ce91
commit e4ec4d2526
2 changed files with 46 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
mediapipe/calculators/util/landmarks_to_render_data_calculator.cc
View File

@ -322,27 +322,30 @@ absl::Status LandmarksToRenderDataCalculator::Process(CalculatorContext* cc) {
options_.presence_threshold(), options_.connection_color(), thickness, options_.presence_threshold(), options_.connection_color(), thickness,
/*normalized=*/false, render_data.get()); /*normalized=*/false, render_data.get());
} }
for (int i = 0; i < landmarks.landmark_size(); ++i) { if (options_.render_landmarks()) {
const Landmark& landmark = landmarks.landmark(i); for (int i = 0; i < landmarks.landmark_size(); ++i) {
const Landmark& landmark = landmarks.landmark(i);
if (!IsLandmarkVisibleAndPresent<Landmark>( if (!IsLandmarkVisibleAndPresent<Landmark>(
landmark, options_.utilize_visibility(), landmark, options_.utilize_visibility(),
options_.visibility_threshold(), options_.utilize_presence(), options_.visibility_threshold(), options_.utilize_presence(),
options_.presence_threshold())) { options_.presence_threshold())) {
continue; continue;
} }
auto* landmark_data_render = AddPointRenderData( auto* landmark_data_render = AddPointRenderData(
options_.landmark_color(), thickness, render_data.get()); options_.landmark_color(), thickness, render_data.get());
if (visualize_depth) { if (visualize_depth) {
SetColorSizeValueFromZ(landmark.z(), z_min, z_max, landmark_data_render, SetColorSizeValueFromZ(landmark.z(), z_min, z_max,
options_.min_depth_circle_thickness(), landmark_data_render,
options_.max_depth_circle_thickness()); options_.min_depth_circle_thickness(),
options_.max_depth_circle_thickness());
}
auto* landmark_data = landmark_data_render->mutable_point();
landmark_data->set_normalized(false);
landmark_data->set_x(landmark.x());
landmark_data->set_y(landmark.y());
} }
auto* landmark_data = landmark_data_render->mutable_point();
landmark_data->set_normalized(false);
landmark_data->set_x(landmark.x());
landmark_data->set_y(landmark.y());
} }
} }
@ -368,27 +371,30 @@ absl::Status LandmarksToRenderDataCalculator::Process(CalculatorContext* cc) {
options_.presence_threshold(), options_.connection_color(), thickness, options_.presence_threshold(), options_.connection_color(), thickness,
/*normalized=*/true, render_data.get()); /*normalized=*/true, render_data.get());
} }
for (int i = 0; i < landmarks.landmark_size(); ++i) { if (options_.render_landmarks()) {
const NormalizedLandmark& landmark = landmarks.landmark(i); for (int i = 0; i < landmarks.landmark_size(); ++i) {
const NormalizedLandmark& landmark = landmarks.landmark(i);
if (!IsLandmarkVisibleAndPresent<NormalizedLandmark>( if (!IsLandmarkVisibleAndPresent<NormalizedLandmark>(
landmark, options_.utilize_visibility(), landmark, options_.utilize_visibility(),
options_.visibility_threshold(), options_.utilize_presence(), options_.visibility_threshold(), options_.utilize_presence(),
options_.presence_threshold())) { options_.presence_threshold())) {
continue; continue;
} }
auto* landmark_data_render = AddPointRenderData( auto* landmark_data_render = AddPointRenderData(
options_.landmark_color(), thickness, render_data.get()); options_.landmark_color(), thickness, render_data.get());
if (visualize_depth) { if (visualize_depth) {
SetColorSizeValueFromZ(landmark.z(), z_min, z_max, landmark_data_render, SetColorSizeValueFromZ(landmark.z(), z_min, z_max,
options_.min_depth_circle_thickness(), landmark_data_render,
options_.max_depth_circle_thickness()); options_.min_depth_circle_thickness(),
options_.max_depth_circle_thickness());
}
auto* landmark_data = landmark_data_render->mutable_point();
landmark_data->set_normalized(true);
landmark_data->set_x(landmark.x());
landmark_data->set_y(landmark.y());
} }
auto* landmark_data = landmark_data_render->mutable_point();
landmark_data->set_normalized(true);
landmark_data->set_x(landmark.x());
landmark_data->set_y(landmark.y());
} }
} }

4
mediapipe/calculators/util/landmarks_to_render_data_calculator.proto
View File

@ -32,6 +32,10 @@ message LandmarksToRenderDataCalculatorOptions {
// Color of the landmarks. // Color of the landmarks.
optional Color landmark_color = 2; optional Color landmark_color = 2;
// Whether to render landmarks as points.
optional bool render_landmarks = 14 [default = true];
// Color of the connections. // Color of the connections.
optional Color connection_color = 3; optional Color connection_color = 3;