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

added smooth landmarks support

Browse Source
This commit is contained in:
Mautisim Munir 2022-09-26 22:26:36 +05:00
parent 65a944e3f9
commit c9670669b2
4 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mediapipe/examples/android/src/java/com/google/mediapipe/apps/posetrackingsolutiongpu/MainActivity.java
View File

@ -275,6 +275,7 @@ public class MainActivity extends AppCompatActivity {
.setStaticImageMode(false) .setStaticImageMode(false)
.setLandmarkVisibility(true) .setLandmarkVisibility(true)
.setModelComplexity(0) .setModelComplexity(0)
.setSmoothLandmarks(true)
.build()); .build());
poseTracking.setErrorListener( poseTracking.setErrorListener(
(message, e) -> Log.e(TAG, "MediaPipe Face Detection error:" + message)); (message, e) -> Log.e(TAG, "MediaPipe Face Detection error:" + message));

1
mediapipe/graphs/pose_tracking/pose_tracking_gpu_image.pbtxt
View File

@ -63,6 +63,7 @@ node {
calculator: "PoseLandmarkGpu" calculator: "PoseLandmarkGpu"
input_side_packet: "ENABLE_SEGMENTATION:enable_segmentation" input_side_packet: "ENABLE_SEGMENTATION:enable_segmentation"
input_side_packet: "MODEL_COMPLEXITY:model_complexity" input_side_packet: "MODEL_COMPLEXITY:model_complexity"
input_side_packet: "SMOOTH_LANDMARKS:smooth_landmarks"
input_stream: "IMAGE:throttled_input_video" input_stream: "IMAGE:throttled_input_video"
output_stream: "LANDMARKS:pose_landmarks" output_stream: "LANDMARKS:pose_landmarks"
output_stream: "SEGMENTATION_MASK:segmentation_mask" output_stream: "SEGMENTATION_MASK:segmentation_mask"

1
mediapipe/java/com/google/mediapipe/solutions/posetracking/PoseTracking.java
View File

@ -128,6 +128,7 @@ public class PoseTracking extends ImageSolutionBase {
Map<String, Packet> inputSidePackets = new HashMap<>(); Map<String, Packet> inputSidePackets = new HashMap<>();
// inputSidePackets.put("enable_segmentation", packetCreator.createBool(false)); // inputSidePackets.put("enable_segmentation", packetCreator.createBool(false));
inputSidePackets.put("model_complexity",packetCreator.createInt32(options.modelComplexity())); inputSidePackets.put("model_complexity",packetCreator.createInt32(options.modelComplexity()));
inputSidePackets.put("smooth_landmarks",packetCreator.createBool(options.smoothLandmarks()));
start(inputSidePackets); start(inputSidePackets);
} }

8
mediapipe/java/com/google/mediapipe/solutions/posetracking/PoseTrackingOptions.java
View File

@ -41,6 +41,8 @@ public abstract class PoseTrackingOptions {
public abstract boolean landmarkVisibility(); public abstract boolean landmarkVisibility();
public abstract boolean smoothLandmarks();
public static Builder builder() { public static Builder builder() {
return new AutoValue_PoseTrackingOptions.Builder().withDefaultValues(); return new AutoValue_PoseTrackingOptions.Builder().withDefaultValues();
} }
@ -49,7 +51,10 @@ public abstract class PoseTrackingOptions {
@AutoValue.Builder @AutoValue.Builder
public abstract static class Builder { public abstract static class Builder {
public Builder withDefaultValues() { public Builder withDefaultValues() {
return setStaticImageMode(false).setModelComplexity(0).setMinDetectionConfidence(0.5f); return setStaticImageMode(false)
.setModelComplexity(0)
.setMinDetectionConfidence(0.5f)
.setSmoothLandmarks(true);
} }
public abstract Builder setStaticImageMode(boolean value); public abstract Builder setStaticImageMode(boolean value);
@ -60,6 +65,7 @@ public abstract class PoseTrackingOptions {
public abstract Builder setLandmarkVisibility(boolean value); public abstract Builder setLandmarkVisibility(boolean value);
public abstract Builder setSmoothLandmarks(boolean value);
public abstract PoseTrackingOptions build(); public abstract PoseTrackingOptions build();
} }
} }