talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: 1e13be30e2c6838d4a2ff768a39c414bc80534bb
This commit is contained in:
MediaPipe Team
2022-09-06 21:46:17 +00:00
committed by Sebastian Schmidt
parent 63e679d99c
commit 4dc4b19ddb
639 changed files with 71327 additions and 2078 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/examples/desktop/autoflip/BUILD
+14 -2
View File
@@ -16,13 +16,22 @@ load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library"
licenses(["notice"])
package(default_visibility = ["//mediapipe/examples:__subpackages__"])
package(default_visibility = [
"//mediapipe/examples:__subpackages__",
])
proto_library(
name = "autoflip_messages_proto",
srcs = ["autoflip_messages.proto"],
deps = [
"//mediapipe/framework:calculator_proto",
"//mediapipe/framework:calculator_options_proto",
],
)
java_lite_proto_library(
name = "autoflip_messages_java_proto_lite",
deps = [
":autoflip_messages_proto",
],
)
@@ -38,6 +47,9 @@ mediapipe_cc_proto_library(
cc_binary(
name = "run_autoflip",
data = [
"//mediapipe/modules/face_detection:face_detection_full_range_sparse.tflite",
],
deps = [
"//mediapipe/calculators/core:packet_thinner_calculator",
"//mediapipe/calculators/image:scale_image_calculator",
mediapipe/examples/desktop/autoflip/autoflip_messages.proto
+3 -1
View File
@@ -17,7 +17,9 @@ syntax = "proto2";
package mediapipe.autoflip;
import "mediapipe/framework/calculator.proto";
import "mediapipe/framework/calculator_options.proto";
option java_multiple_files = true;
// Borders detected on the frame as well as non-border color (if present).
// Next tag: 4
mediapipe/examples/desktop/autoflip/calculators/BUILD
+1 -9
View File
@@ -289,7 +289,6 @@ cc_library(
":signal_fusing_calculator_cc_proto",
"//mediapipe/examples/desktop/autoflip:autoflip_messages_cc_proto",
"//mediapipe/framework:calculator_framework",
"//mediapipe/framework/formats:image_frame",
"//mediapipe/framework/port:ret_check",
"//mediapipe/framework/port:status",
"@com_google_absl//absl/container:btree",
@@ -343,7 +342,6 @@ cc_library(
visibility = ["//visibility:public"],
deps = [
":shot_boundary_calculator_cc_proto",
"//mediapipe/examples/desktop/autoflip:autoflip_messages_cc_proto",
"//mediapipe/framework:calculator_framework",
"//mediapipe/framework:timestamp",
"//mediapipe/framework/formats:image_frame",
@@ -358,10 +356,7 @@ cc_library(
proto_library(
name = "shot_boundary_calculator_proto",
srcs = ["shot_boundary_calculator.proto"],
deps = [
"//mediapipe/examples/desktop/autoflip:autoflip_messages_proto",
"//mediapipe/framework:calculator_proto",
],
deps = ["//mediapipe/framework:calculator_proto"],
)
mediapipe_cc_proto_library(
@@ -414,7 +409,6 @@ cc_library(
"//mediapipe/framework/port:opencv_imgproc",
"//mediapipe/framework/port:ret_check",
"//mediapipe/framework/port:status",
"@com_google_absl//absl/memory",
],
alwayslink = 1,
)
@@ -452,7 +446,6 @@ cc_test(
"//mediapipe/framework/formats:detection_cc_proto",
"//mediapipe/framework/formats:image_frame",
"//mediapipe/framework/formats:image_frame_opencv",
"//mediapipe/framework/formats:location_data_cc_proto",
"//mediapipe/framework/port:gtest_main",
"//mediapipe/framework/port:parse_text_proto",
"//mediapipe/framework/port:ret_check",
@@ -505,7 +498,6 @@ cc_test(
"//mediapipe/framework:calculator_framework",
"//mediapipe/framework:calculator_runner",
"//mediapipe/framework/formats:detection_cc_proto",
"//mediapipe/framework/formats:location_data_cc_proto",
"//mediapipe/framework/port:gtest_main",
"//mediapipe/framework/port:parse_text_proto",
"//mediapipe/framework/port:ret_check",
mediapipe/examples/desktop/autoflip/calculators/content_zooming_calculator.cc
+1
View File
@@ -203,6 +203,7 @@ absl::Status ContentZoomingCalculator::GetContract(
}
absl::Status ContentZoomingCalculator::Open(mediapipe::CalculatorContext* cc) {
cc->SetOffset(mediapipe::TimestampDiff(0));
options_ = cc->Options<ContentZoomingCalculatorOptions>();
if (options_.has_kinematic_options()) {
return mediapipe::UnknownErrorBuilder(MEDIAPIPE_LOC)
mediapipe/examples/desktop/autoflip/quality/BUILD
+3 -1
View File
@@ -16,7 +16,9 @@ load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library"
licenses(["notice"])
package(default_visibility = ["//mediapipe/examples:__subpackages__"])
package(default_visibility = [
"//mediapipe/examples:__subpackages__",
])
proto_library(
name = "cropping_proto",
mediapipe/examples/desktop/face_detection/BUILD
+3
View File
@@ -18,6 +18,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "face_detection_full_range_cpu",
data = ["//mediapipe/modules/face_detection:face_detection_full_range.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/face_detection:face_detection_full_range_desktop_live_deps",
@@ -26,6 +27,7 @@ cc_binary(
cc_binary(
name = "face_detection_cpu",
data = ["//mediapipe/modules/face_detection:face_detection_short_range.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/face_detection:desktop_live_calculators",
@@ -35,6 +37,7 @@ cc_binary(
# Linux only
cc_binary(
name = "face_detection_gpu",
data = ["//mediapipe/modules/face_detection:face_detection_short_range.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/face_detection:desktop_live_gpu_calculators",
mediapipe/examples/desktop/face_mesh/BUILD
+3
View File
@@ -18,6 +18,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "face_mesh_tflite",
data = ["//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite"],
deps = [
"//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/face_mesh:desktop_calculators",
@@ -26,6 +27,7 @@ cc_binary(
cc_binary(
name = "face_mesh_cpu",
data = ["//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/face_mesh:desktop_live_calculators",
@@ -35,6 +37,7 @@ cc_binary(
# Linux only
cc_binary(
name = "face_mesh_gpu",
data = ["//mediapipe/modules/face_landmark:face_landmark_with_attention.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/face_mesh:desktop_live_gpu_calculators",
mediapipe/examples/desktop/hair_segmentation/BUILD
+2
View File
@@ -19,6 +19,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
# Linux only
cc_binary(
name = "hair_segmentation_gpu",
data = ["//mediapipe/models:hair_segmentation.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/hair_segmentation:mobile_calculators",
@@ -27,6 +28,7 @@ cc_binary(
cc_binary(
name = "hair_segmentation_cpu",
data = ["//mediapipe/models:hair_segmentation.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
] + select({
mediapipe/examples/desktop/hand_tracking/BUILD
+12
View File
@@ -18,6 +18,10 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "hand_tracking_tflite",
data = [
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/palm_detection:palm_detection_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/hand_tracking:desktop_tflite_calculators",
@@ -26,6 +30,10 @@ cc_binary(
cc_binary(
name = "hand_tracking_cpu",
data = [
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/palm_detection:palm_detection_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/hand_tracking:desktop_tflite_calculators",
@@ -35,6 +43,10 @@ cc_binary(
# Linux only
cc_binary(
name = "hand_tracking_gpu",
data = [
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/palm_detection:palm_detection_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/hand_tracking:mobile_calculators",
mediapipe/examples/desktop/holistic_tracking/BUILD
+14
View File
@@ -18,6 +18,13 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "holistic_tracking_cpu",
data = [
"//mediapipe/modules/face_landmark:face_landmark.tflite",
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/holistic_landmark:hand_recrop.tflite",
"//mediapipe/modules/pose_detection:pose_detection.tflite",
"//mediapipe/modules/pose_landmark:pose_landmark_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/holistic_tracking:holistic_tracking_cpu_graph_deps",
@@ -27,6 +34,13 @@ cc_binary(
# Linux only
cc_binary(
name = "holistic_tracking_gpu",
data = [
"//mediapipe/modules/face_landmark:face_landmark.tflite",
"//mediapipe/modules/hand_landmark:hand_landmark_full.tflite",
"//mediapipe/modules/holistic_landmark:hand_recrop.tflite",
"//mediapipe/modules/pose_detection:pose_detection.tflite",
"//mediapipe/modules/pose_landmark:pose_landmark_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/holistic_tracking:holistic_tracking_gpu_deps",
mediapipe/examples/desktop/iris_tracking/BUILD
+4
View File
@@ -19,6 +19,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "iris_depth_from_image_desktop",
srcs = ["iris_depth_from_image_desktop.cc"],
data = ["//mediapipe/modules/iris_landmark:iris_landmark.tflite"],
deps = [
"//mediapipe/framework:calculator_framework",
"//mediapipe/framework/formats:image_frame",
@@ -37,6 +38,7 @@ cc_binary(
cc_binary(
name = "iris_tracking_cpu_video_input",
data = ["//mediapipe/modules/iris_landmark:iris_landmark.tflite"],
deps = [
"//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/iris_tracking:iris_tracking_cpu_video_input_deps",
@@ -45,6 +47,7 @@ cc_binary(
cc_binary(
name = "iris_tracking_cpu",
data = ["//mediapipe/modules/iris_landmark:iris_landmark.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/iris_tracking:iris_tracking_cpu_deps",
@@ -54,6 +57,7 @@ cc_binary(
# Linux only
cc_binary(
name = "iris_tracking_gpu",
data = ["//mediapipe/modules/iris_landmark:iris_landmark.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/iris_tracking:iris_tracking_gpu_deps",
mediapipe/examples/desktop/object_detection/BUILD
+12
View File
@@ -18,6 +18,10 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "object_detection_tensorflow",
data = [
"//mediapipe/models:ssdlite_object_detection.tflite",
"//mediapipe/models:ssdlite_object_detection_labelmap.txt",
],
deps = [
"//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/object_detection:desktop_tensorflow_calculators",
@@ -28,6 +32,10 @@ cc_binary(
cc_binary(
name = "object_detection_tflite",
data = [
"//mediapipe/models:ssdlite_object_detection.tflite",
"//mediapipe/models:ssdlite_object_detection_labelmap.txt",
],
deps = [
"//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/object_detection:desktop_tflite_calculators",
@@ -36,6 +44,10 @@ cc_binary(
cc_binary(
name = "object_detection_cpu",
data = [
"//mediapipe/models:ssdlite_object_detection.tflite",
"//mediapipe/models:ssdlite_object_detection_labelmap.txt",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/object_detection:desktop_tflite_calculators",
mediapipe/examples/desktop/object_detection_3d/BUILD
+9
View File
@@ -27,6 +27,15 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
# Cup: box_landmark_model_path=mediapipe/modules/objectron/object_detection_3d_cup.tflite,allowed_labels=Mug
cc_binary(
name = "objectron_cpu",
data = [
"//mediapipe/modules/objectron:object_detection_3d_camera.tflite",
"//mediapipe/modules/objectron:object_detection_3d_chair.tflite",
"//mediapipe/modules/objectron:object_detection_3d_chair_1stage.tflite",
"//mediapipe/modules/objectron:object_detection_3d_cup.tflite",
"//mediapipe/modules/objectron:object_detection_3d_sneakers.tflite",
"//mediapipe/modules/objectron:object_detection_3d_sneakers_1stage.tflite",
"//mediapipe/modules/objectron:object_detection_ssd_mobilenetv2_oidv4_fp16.tflite",
],
deps = [
"//mediapipe/examples/desktop:simple_run_graph_main",
"//mediapipe/graphs/object_detection_3d:desktop_cpu_calculators",
mediapipe/examples/desktop/object_tracking/BUILD
+4
View File
@@ -18,6 +18,10 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "object_tracking_cpu",
data = [
"//mediapipe/models:ssdlite_object_detection.tflite",
"//mediapipe/models:ssdlite_object_detection_labelmap.txt",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/tracking:desktop_calculators",
mediapipe/examples/desktop/pose_tracking/BUILD
+8
View File
@@ -18,6 +18,10 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "pose_tracking_cpu",
data = [
"//mediapipe/modules/pose_detection:pose_detection.tflite",
"//mediapipe/modules/pose_landmark:pose_landmark_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/pose_tracking:pose_tracking_cpu_deps",
@@ -27,6 +31,10 @@ cc_binary(
# Linux only
cc_binary(
name = "pose_tracking_gpu",
data = [
"//mediapipe/modules/pose_detection:pose_detection.tflite",
"//mediapipe/modules/pose_landmark:pose_landmark_full.tflite",
],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/pose_tracking:pose_tracking_gpu_deps",
mediapipe/examples/desktop/selfie_segmentation/BUILD
+2
View File
@@ -18,6 +18,7 @@ package(default_visibility = ["//mediapipe/examples:__subpackages__"])
cc_binary(
name = "selfie_segmentation_cpu",
data = ["//mediapipe/modules/selfie_segmentation:selfie_segmentation.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main",
"//mediapipe/graphs/selfie_segmentation:selfie_segmentation_cpu_deps",
@@ -27,6 +28,7 @@ cc_binary(
# Linux only
cc_binary(
name = "selfie_segmentation_gpu",
data = ["//mediapipe/modules/selfie_segmentation:selfie_segmentation.tflite"],
deps = [
"//mediapipe/examples/desktop:demo_run_graph_main_gpu",
"//mediapipe/graphs/selfie_segmentation:selfie_segmentation_gpu_deps",