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

changed timestamp to nanoseconds, added presence

Browse Source
This commit is contained in:
Mautisim Munir 2022-10-28 14:01:10 +05:00
parent 0d907a79d1
commit fa5da361d7
3 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/java/com/google/mediapipe/solutions/lindera/ComputerVisionPlugin.java
View File

@ -1,5 +1,5 @@
package com.google.mediapipe.solutions.lindera; package com.google.mediapipe.solutions.lindera;
public interface ComputerVisionPlugin { public interface ComputerVisionPlugin {
void bodyJoints(int timestamp, BodyJoints bodyJoints); void bodyJoints(long timestamp, BodyJoints bodyJoints);
} }

12
mediapipe/java/com/google/mediapipe/solutions/lindera/Lindera.java
View File

@ -25,8 +25,10 @@ public class Lindera {
public FpsHelper fpsHelper = new FpsHelper(); public FpsHelper fpsHelper = new FpsHelper();
private PoseTracking poseTracking; private PoseTracking poseTracking;
// TODO: Verify that this is the timestamp used in Actual Plugin
private int timeStamp = 0; private long getTimeStamp(){
return System.nanoTime();
}
private CameraRotation cameraRotation = CameraRotation.AUTOMATIC; private CameraRotation cameraRotation = CameraRotation.AUTOMATIC;
// Live camera demo UI and camera components. // Live camera demo UI and camera components.
@ -85,14 +87,13 @@ public class Lindera {
glSurfaceView.setRenderData(poseTrackingResult); glSurfaceView.setRenderData(poseTrackingResult);
glSurfaceView.requestRender(); glSurfaceView.requestRender();
ImmutableList<LandmarkProto.Landmark> landmarks = poseTrackingResult.multiPoseLandmarks(); ImmutableList<LandmarkProto.Landmark> landmarks = poseTrackingResult.multiPoseLandmarks();
timeStamp+=1;
if (landmarks.isEmpty()) return; if (landmarks.isEmpty()) return;
BodyJoints bodyJoints = new BodyJoints(); BodyJoints bodyJoints = new BodyJoints();
landmarksToBodyJoints(landmarks,bodyJoints); landmarksToBodyJoints(landmarks,bodyJoints);
plugin.bodyJoints(timeStamp, bodyJoints); plugin.bodyJoints(getTimeStamp(), bodyJoints);
}); });
} }
@ -180,7 +181,6 @@ public class Lindera {
if (poseTracking != null) { if (poseTracking != null) {
poseTracking.close(); poseTracking.close();
} }
timeStamp = 0;
} }
private void landmarkToXYZPointWithConfidence(LandmarkProto.Landmark landmark,XYZPointWithConfidence bodyJoint){ private void landmarkToXYZPointWithConfidence(LandmarkProto.Landmark landmark,XYZPointWithConfidence bodyJoint){
@ -188,6 +188,8 @@ public class Lindera {
bodyJoint.y = landmark.getY(); bodyJoint.y = landmark.getY();
bodyJoint.z = landmark.getZ(); bodyJoint.z = landmark.getZ();
bodyJoint.confidence = landmark.getVisibility(); bodyJoint.confidence = landmark.getVisibility();
bodyJoint.presence = landmark.getPresence();
} }
private void landmarksToBodyJoints(ImmutableList<LandmarkProto.Landmark> landmarks , BodyJoints bodyJoints){ private void landmarksToBodyJoints(ImmutableList<LandmarkProto.Landmark> landmarks , BodyJoints bodyJoints){

1
mediapipe/java/com/google/mediapipe/solutions/lindera/XYZPointWithConfidence.java
View File

@ -5,6 +5,7 @@ public class XYZPointWithConfidence {
public float y = 0; public float y = 0;
public float z = 0; public float z = 0;
public float confidence = 0; public float confidence = 0;
public float presence = 0;