set different graph for solution api

This commit is contained in:
Mautisim Munir 2022-09-26 17:39:26 +05:00
parent 065571100c
commit cb9c9e4712
2 changed files with 2 additions and 2 deletions

View File

@ -27,7 +27,7 @@ android_library(
# "//mediapipe/modules/face_detection:face_detection_full_range_sparse.tflite", # "//mediapipe/modules/face_detection:face_detection_full_range_sparse.tflite",
# "//mediapipe/modules/face_detection:face_detection_short_range.tflite", # "//mediapipe/modules/face_detection:face_detection_short_range.tflite",
# "//mediapipe/modules/face_detection:face_detection_short_range_image.binarypb", # "//mediapipe/modules/face_detection:face_detection_short_range_image.binarypb",
"//mediapipe/graphs/pose_tracking:pose_tracking_gpu.binarypb", "//mediapipe/graphs/pose_tracking:pose_tracking_gpu_image.binarypb",
"//mediapipe/modules/pose_landmark:pose_landmark_heavy.tflite", "//mediapipe/modules/pose_landmark:pose_landmark_heavy.tflite",
"//mediapipe/modules/pose_landmark:pose_landmark_full.tflite", "//mediapipe/modules/pose_landmark:pose_landmark_full.tflite",
"//mediapipe/modules/pose_landmark:pose_landmark_lite.tflite", "//mediapipe/modules/pose_landmark:pose_landmark_lite.tflite",

View File

@ -44,7 +44,7 @@ import javax.annotation.Nullable;
public class PoseTracking extends ImageSolutionBase { public class PoseTracking extends ImageSolutionBase {
private static final String TAG = "PoseTracking"; private static final String TAG = "PoseTracking";
private static final String SHORT_RANGE_GRAPH_NAME = "pose_tracking_gpu.binarypb"; private static final String SHORT_RANGE_GRAPH_NAME = "pose_tracking_gpu_image.binarypb";
private static final String FULL_RANGE_GRAPH_NAME = "face_detection_full_range_image.binarypb"; private static final String FULL_RANGE_GRAPH_NAME = "face_detection_full_range_image.binarypb";
private static final String IMAGE_INPUT_STREAM = "input_video"; private static final String IMAGE_INPUT_STREAM = "input_video";
private static final ImmutableList<String> OUTPUT_STREAMS = private static final ImmutableList<String> OUTPUT_STREAMS =