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

Internal change

Browse Source 
PiperOrigin-RevId: 520486540
This commit is contained in:
Sebastian Schmidt 2023-03-29 17:24:52 -07:00 committed by Copybara-Service
parent 6be4aedcf7
commit df67b35348
9 changed files with 421 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mediapipe/calculators/internal/BUILD
View File

@ -12,20 +12,25 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
licenses(["notice"]) licenses(["notice"])
package(default_visibility = ["//visibility:private"]) package(default_visibility = ["//visibility:private"])
mediapipe_proto_library( proto_library(
name = "callback_packet_calculator_proto", name = "callback_packet_calculator_proto",
srcs = ["callback_packet_calculator.proto"], srcs = ["callback_packet_calculator.proto"],
visibility = ["//mediapipe/framework:__subpackages__"], visibility = ["//mediapipe/framework:__subpackages__"],
deps = [ deps = ["//mediapipe/framework:calculator_proto"],
"//mediapipe/framework:calculator_options_proto", )
"//mediapipe/framework:calculator_proto",
], mediapipe_cc_proto_library(
name = "callback_packet_calculator_cc_proto",
srcs = ["callback_packet_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
visibility = ["//mediapipe/framework:__subpackages__"],
deps = [":callback_packet_calculator_proto"],
) )
cc_library( cc_library(

106
mediapipe/calculators/video/BUILD
View File

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
# #
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
load( load(
"//mediapipe/framework/tool:mediapipe_graph.bzl", "//mediapipe/framework/tool:mediapipe_graph.bzl",
"mediapipe_binary_graph", "mediapipe_binary_graph",
@ -23,35 +23,28 @@ licenses(["notice"])
package(default_visibility = ["//visibility:public"]) package(default_visibility = ["//visibility:public"])
mediapipe_proto_library( proto_library(
name = "flow_to_image_calculator_proto", name = "flow_to_image_calculator_proto",
srcs = ["flow_to_image_calculator.proto"], srcs = ["flow_to_image_calculator.proto"],
deps = [ deps = ["//mediapipe/framework:calculator_proto"],
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto",
],
) )
mediapipe_proto_library( proto_library(
name = "opencv_video_encoder_calculator_proto", name = "opencv_video_encoder_calculator_proto",
srcs = ["opencv_video_encoder_calculator.proto"], srcs = ["opencv_video_encoder_calculator.proto"],
deps = [ deps = ["//mediapipe/framework:calculator_proto"],
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto",
],
) )
mediapipe_proto_library( proto_library(
name = "motion_analysis_calculator_proto", name = "motion_analysis_calculator_proto",
srcs = ["motion_analysis_calculator.proto"], srcs = ["motion_analysis_calculator.proto"],
deps = [ deps = [
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
"//mediapipe/util/tracking:motion_analysis_proto", "//mediapipe/util/tracking:motion_analysis_proto",
], ],
) )
mediapipe_proto_library( proto_library(
name = "flow_packager_calculator_proto", name = "flow_packager_calculator_proto",
srcs = ["flow_packager_calculator.proto"], srcs = ["flow_packager_calculator.proto"],
deps = [ deps = [
@ -60,45 +53,114 @@ mediapipe_proto_library(
], ],
) )
mediapipe_proto_library( proto_library(
name = "box_tracker_calculator_proto", name = "box_tracker_calculator_proto",
srcs = ["box_tracker_calculator.proto"], srcs = ["box_tracker_calculator.proto"],
deps = [ deps = [
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
"//mediapipe/util/tracking:box_tracker_proto", "//mediapipe/util/tracking:box_tracker_proto",
], ],
) )
mediapipe_proto_library( proto_library(
name = "tracked_detection_manager_calculator_proto", name = "tracked_detection_manager_calculator_proto",
srcs = ["tracked_detection_manager_calculator.proto"], srcs = ["tracked_detection_manager_calculator.proto"],
deps = [ deps = [
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
"//mediapipe/util/tracking:tracked_detection_manager_config_proto", "//mediapipe/util/tracking:tracked_detection_manager_config_proto",
], ],
) )
mediapipe_proto_library( proto_library(
name = "box_detector_calculator_proto", name = "box_detector_calculator_proto",
srcs = ["box_detector_calculator.proto"], srcs = ["box_detector_calculator.proto"],
deps = [ deps = [
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
"//mediapipe/util/tracking:box_detector_proto", "//mediapipe/util/tracking:box_detector_proto",
], ],
) )
mediapipe_proto_library( proto_library(
name = "video_pre_stream_calculator_proto", name = "video_pre_stream_calculator_proto",
srcs = ["video_pre_stream_calculator.proto"], srcs = ["video_pre_stream_calculator.proto"],
deps = [ deps = [
"//mediapipe/framework:calculator_options_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
], ],
) )
mediapipe_cc_proto_library(
name = "motion_analysis_calculator_cc_proto",
srcs = ["motion_analysis_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/util/tracking:motion_analysis_cc_proto",
],
deps = [":motion_analysis_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "flow_packager_calculator_cc_proto",
srcs = ["flow_packager_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/util/tracking:flow_packager_cc_proto",
],
deps = [":flow_packager_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "box_tracker_calculator_cc_proto",
srcs = ["box_tracker_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/util/tracking:box_tracker_cc_proto",
],
deps = [":box_tracker_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "tracked_detection_manager_calculator_cc_proto",
srcs = ["tracked_detection_manager_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/util/tracking:tracked_detection_manager_config_cc_proto",
],
deps = [":tracked_detection_manager_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "box_detector_calculator_cc_proto",
srcs = ["box_detector_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/util/tracking:box_detector_cc_proto",
],
deps = [":box_detector_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "video_pre_stream_calculator_cc_proto",
srcs = ["video_pre_stream_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
],
deps = [":video_pre_stream_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "flow_to_image_calculator_cc_proto",
srcs = ["flow_to_image_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
deps = [":flow_to_image_calculator_proto"],
)
mediapipe_cc_proto_library(
name = "opencv_video_encoder_calculator_cc_proto",
srcs = ["opencv_video_encoder_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
deps = [":opencv_video_encoder_calculator_proto"],
)
cc_library( cc_library(
name = "flow_to_image_calculator", name = "flow_to_image_calculator",
srcs = ["flow_to_image_calculator.cc"], srcs = ["flow_to_image_calculator.cc"],

104
mediapipe/examples/desktop/autoflip/calculators/BUILD
View File

@ -1,4 +1,4 @@
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
# Copyright 2019 The MediaPipe Authors. # Copyright 2019 The MediaPipe Authors.
# #
@ -40,15 +40,22 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "border_detection_calculator_proto", name = "border_detection_calculator_proto",
srcs = ["border_detection_calculator.proto"], srcs = ["border_detection_calculator.proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [ deps = [
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
], ],
) )
mediapipe_cc_proto_library(
name = "border_detection_calculator_cc_proto",
srcs = ["border_detection_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [":border_detection_calculator_proto"],
)
cc_library( cc_library(
name = "content_zooming_calculator_state", name = "content_zooming_calculator_state",
hdrs = ["content_zooming_calculator_state.h"], hdrs = ["content_zooming_calculator_state.h"],
@ -78,15 +85,27 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "content_zooming_calculator_proto", name = "content_zooming_calculator_proto",
srcs = ["content_zooming_calculator.proto"], srcs = ["content_zooming_calculator.proto"],
deps = [
"//mediapipe/examples/desktop/autoflip/quality:kinematic_path_solver_proto",
"//mediapipe/framework:calculator_proto",
],
)
mediapipe_cc_proto_library(
name = "content_zooming_calculator_cc_proto",
srcs = ["content_zooming_calculator.proto"],
cc_deps = [
"//mediapipe/examples/desktop/autoflip/quality:kinematic_path_solver_cc_proto",
"//mediapipe/framework:calculator_cc_proto",
],
visibility = [ visibility = [
"//mediapipe/examples:__subpackages__", "//mediapipe/examples:__subpackages__",
], ],
deps = [ deps = [
"//mediapipe/examples/desktop/autoflip/quality:kinematic_path_solver_proto", ":content_zooming_calculator_proto",
"//mediapipe/framework:calculator_proto",
], ],
) )
@ -158,7 +177,7 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "video_filtering_calculator_proto", name = "video_filtering_calculator_proto",
srcs = ["video_filtering_calculator.proto"], srcs = ["video_filtering_calculator.proto"],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
@ -167,6 +186,14 @@ mediapipe_proto_library(
], ],
) )
mediapipe_cc_proto_library(
name = "video_filtering_calculator_cc_proto",
srcs = ["video_filtering_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
visibility = ["//visibility:public"],
deps = [":video_filtering_calculator_proto"],
)
cc_test( cc_test(
name = "video_filtering_calculator_test", name = "video_filtering_calculator_test",
srcs = ["video_filtering_calculator_test.cc"], srcs = ["video_filtering_calculator_test.cc"],
@ -182,7 +209,7 @@ cc_test(
], ],
) )
mediapipe_proto_library( proto_library(
name = "scene_cropping_calculator_proto", name = "scene_cropping_calculator_proto",
srcs = ["scene_cropping_calculator.proto"], srcs = ["scene_cropping_calculator.proto"],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
@ -192,6 +219,17 @@ mediapipe_proto_library(
], ],
) )
mediapipe_cc_proto_library(
name = "scene_cropping_calculator_cc_proto",
srcs = ["scene_cropping_calculator.proto"],
cc_deps = [
"//mediapipe/examples/desktop/autoflip/quality:cropping_cc_proto",
"//mediapipe/framework:calculator_cc_proto",
],
visibility = ["//visibility:public"],
deps = [":scene_cropping_calculator_proto"],
)
cc_library( cc_library(
name = "scene_cropping_calculator", name = "scene_cropping_calculator",
srcs = ["scene_cropping_calculator.cc"], srcs = ["scene_cropping_calculator.cc"],
@ -258,16 +296,26 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "signal_fusing_calculator_proto", name = "signal_fusing_calculator_proto",
srcs = ["signal_fusing_calculator.proto"], srcs = ["signal_fusing_calculator.proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [ deps = [
"//mediapipe/examples/desktop/autoflip:autoflip_messages_proto", "//mediapipe/examples/desktop/autoflip:autoflip_messages_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
], ],
) )
mediapipe_cc_proto_library(
name = "signal_fusing_calculator_cc_proto",
srcs = ["signal_fusing_calculator.proto"],
cc_deps = [
"//mediapipe/examples/desktop/autoflip:autoflip_messages_cc_proto",
"//mediapipe/framework:calculator_cc_proto",
],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [":signal_fusing_calculator_proto"],
)
cc_test( cc_test(
name = "signal_fusing_calculator_test", name = "signal_fusing_calculator_test",
srcs = ["signal_fusing_calculator_test.cc"], srcs = ["signal_fusing_calculator_test.cc"],
@ -305,13 +353,20 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "shot_boundary_calculator_proto", name = "shot_boundary_calculator_proto",
srcs = ["shot_boundary_calculator.proto"], srcs = ["shot_boundary_calculator.proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = ["//mediapipe/framework:calculator_proto"], deps = ["//mediapipe/framework:calculator_proto"],
) )
mediapipe_cc_proto_library(
name = "shot_boundary_calculator_cc_proto",
srcs = ["shot_boundary_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [":shot_boundary_calculator_proto"],
)
cc_test( cc_test(
name = "shot_boundary_calculator_test", name = "shot_boundary_calculator_test",
srcs = ["shot_boundary_calculator_test.cc"], srcs = ["shot_boundary_calculator_test.cc"],
@ -358,16 +413,26 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "face_to_region_calculator_proto", name = "face_to_region_calculator_proto",
srcs = ["face_to_region_calculator.proto"], srcs = ["face_to_region_calculator.proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [ deps = [
"//mediapipe/examples/desktop/autoflip/quality:visual_scorer_proto", "//mediapipe/examples/desktop/autoflip/quality:visual_scorer_proto",
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
], ],
) )
mediapipe_cc_proto_library(
name = "face_to_region_calculator_cc_proto",
srcs = ["face_to_region_calculator.proto"],
cc_deps = [
"//mediapipe/examples/desktop/autoflip/quality:visual_scorer_cc_proto",
"//mediapipe/framework:calculator_cc_proto",
],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [":face_to_region_calculator_proto"],
)
cc_test( cc_test(
name = "face_to_region_calculator_test", name = "face_to_region_calculator_test",
srcs = ["face_to_region_calculator_test.cc"], srcs = ["face_to_region_calculator_test.cc"],
@ -389,15 +454,22 @@ cc_test(
], ],
) )
mediapipe_proto_library( proto_library(
name = "localization_to_region_calculator_proto", name = "localization_to_region_calculator_proto",
srcs = ["localization_to_region_calculator.proto"], srcs = ["localization_to_region_calculator.proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [ deps = [
"//mediapipe/framework:calculator_proto", "//mediapipe/framework:calculator_proto",
], ],
) )
mediapipe_cc_proto_library(
name = "localization_to_region_calculator_cc_proto",
srcs = ["localization_to_region_calculator.proto"],
cc_deps = ["//mediapipe/framework:calculator_cc_proto"],
visibility = ["//mediapipe/examples:__subpackages__"],
deps = [":localization_to_region_calculator_proto"],
)
cc_library( cc_library(
name = "localization_to_region_calculator", name = "localization_to_region_calculator",
srcs = ["localization_to_region_calculator.cc"], srcs = ["localization_to_region_calculator.cc"],

10
mediapipe/framework/formats/motion/BUILD
View File

@ -16,17 +16,23 @@
# Description: # Description:
# Working with dense optical flow in mediapipe. # Working with dense optical flow in mediapipe.
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
licenses(["notice"]) licenses(["notice"])
package(default_visibility = ["//visibility:public"]) package(default_visibility = ["//visibility:public"])
mediapipe_proto_library( proto_library(
name = "optical_flow_field_data_proto", name = "optical_flow_field_data_proto",
srcs = ["optical_flow_field_data.proto"], srcs = ["optical_flow_field_data.proto"],
) )
mediapipe_cc_proto_library(
name = "optical_flow_field_data_cc_proto",
srcs = ["optical_flow_field_data.proto"],
deps = [":optical_flow_field_data_proto"],
)
cc_library( cc_library(
name = "optical_flow_field", name = "optical_flow_field",
srcs = ["optical_flow_field.cc"], srcs = ["optical_flow_field.cc"],

42
mediapipe/framework/stream_handler/BUILD
View File

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
# #
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
licenses(["notice"]) licenses(["notice"])
@ -22,32 +22,56 @@ package(
features = ["-layering_check"], features = ["-layering_check"],
) )
mediapipe_proto_library( proto_library(
name = "default_input_stream_handler_proto", name = "default_input_stream_handler_proto",
srcs = ["default_input_stream_handler.proto"], srcs = ["default_input_stream_handler.proto"],
deps = ["//mediapipe/framework:mediapipe_options_proto"], deps = ["//mediapipe/framework:mediapipe_options_proto"],
alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "fixed_size_input_stream_handler_proto", name = "fixed_size_input_stream_handler_proto",
srcs = ["fixed_size_input_stream_handler.proto"], srcs = ["fixed_size_input_stream_handler.proto"],
deps = ["//mediapipe/framework:mediapipe_options_proto"], deps = ["//mediapipe/framework:mediapipe_options_proto"],
alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "sync_set_input_stream_handler_proto", name = "sync_set_input_stream_handler_proto",
srcs = ["sync_set_input_stream_handler.proto"], srcs = ["sync_set_input_stream_handler.proto"],
deps = ["//mediapipe/framework:mediapipe_options_proto"], deps = ["//mediapipe/framework:mediapipe_options_proto"],
alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "timestamp_align_input_stream_handler_proto", name = "timestamp_align_input_stream_handler_proto",
srcs = ["timestamp_align_input_stream_handler.proto"], srcs = ["timestamp_align_input_stream_handler.proto"],
deps = ["//mediapipe/framework:mediapipe_options_proto"], deps = ["//mediapipe/framework:mediapipe_options_proto"],
alwayslink = 1, )
mediapipe_cc_proto_library(
name = "default_input_stream_handler_cc_proto",
srcs = ["default_input_stream_handler.proto"],
cc_deps = ["//mediapipe/framework:mediapipe_options_cc_proto"],
deps = [":default_input_stream_handler_proto"],
)
mediapipe_cc_proto_library(
name = "fixed_size_input_stream_handler_cc_proto",
srcs = ["fixed_size_input_stream_handler.proto"],
cc_deps = ["//mediapipe/framework:mediapipe_options_cc_proto"],
deps = [":fixed_size_input_stream_handler_proto"],
)
mediapipe_cc_proto_library(
name = "sync_set_input_stream_handler_cc_proto",
srcs = ["sync_set_input_stream_handler.proto"],
cc_deps = ["//mediapipe/framework:mediapipe_options_cc_proto"],
deps = [":sync_set_input_stream_handler_proto"],
)
mediapipe_cc_proto_library(
name = "timestamp_align_input_stream_handler_cc_proto",
srcs = ["timestamp_align_input_stream_handler.proto"],
cc_deps = ["//mediapipe/framework:mediapipe_options_cc_proto"],
deps = [":timestamp_align_input_stream_handler_proto"],
) )
cc_library( cc_library(

1
mediapipe/framework/tool/mediapipe_proto.bzl
View File

@ -90,6 +90,7 @@ def mediapipe_proto_library_impl(
visibility = visibility, visibility = visibility,
testonly = testonly, testonly = testonly,
compatible_with = compatible_with, compatible_with = compatible_with,
alwayslink = alwayslink,
)) ))
if def_cc_proto: if def_cc_proto:

27
mediapipe/gpu/BUILD
View File

@ -15,7 +15,7 @@
load("@bazel_skylib//lib:selects.bzl", "selects") load("@bazel_skylib//lib:selects.bzl", "selects")
load("//mediapipe/gpu:metal.bzl", "metal_library") load("//mediapipe/gpu:metal.bzl", "metal_library")
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test") load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library", "mediapipe_proto_library")
load("//mediapipe/framework:mediapipe_cc_test.bzl", "mediapipe_cc_test") load("//mediapipe/framework:mediapipe_cc_test.bzl", "mediapipe_cc_test")
load("//mediapipe/framework:more_selects.bzl", "more_selects") load("//mediapipe/framework:more_selects.bzl", "more_selects")
@ -982,7 +982,7 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "gl_surface_sink_calculator_proto", name = "gl_surface_sink_calculator_proto",
srcs = ["gl_surface_sink_calculator.proto"], srcs = ["gl_surface_sink_calculator.proto"],
deps = [ deps = [
@ -991,6 +991,17 @@ mediapipe_proto_library(
], ],
) )
mediapipe_cc_proto_library(
name = "gl_surface_sink_calculator_cc_proto",
srcs = ["gl_surface_sink_calculator.proto"],
cc_deps = [
":scale_mode_cc_proto",
"//mediapipe/framework:calculator_cc_proto",
],
visibility = ["//visibility:public"],
deps = [":gl_surface_sink_calculator_proto"],
)
### Metal calculators ### Metal calculators
metal_library( metal_library(
@ -1006,13 +1017,23 @@ objc_library(
deps = [":simple_shaders_mtl"], deps = [":simple_shaders_mtl"],
) )
mediapipe_proto_library( proto_library(
name = "copy_calculator_proto", name = "copy_calculator_proto",
srcs = ["copy_calculator.proto"], srcs = ["copy_calculator.proto"],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = ["//mediapipe/framework:calculator_proto"], deps = ["//mediapipe/framework:calculator_proto"],
) )
mediapipe_cc_proto_library(
name = "copy_calculator_cc_proto",
srcs = ["copy_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
],
visibility = ["//visibility:public"],
deps = [":copy_calculator_proto"],
)
objc_library( objc_library(
name = "metal_copy_calculator", name = "metal_copy_calculator",
srcs = ["MetalCopyCalculator.mm"], srcs = ["MetalCopyCalculator.mm"],

28
mediapipe/graphs/iris_tracking/calculators/BUILD
View File

@ -12,11 +12,11 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
licenses(["notice"]) licenses(["notice"])
mediapipe_proto_library( proto_library(
name = "iris_to_render_data_calculator_proto", name = "iris_to_render_data_calculator_proto",
srcs = ["iris_to_render_data_calculator.proto"], srcs = ["iris_to_render_data_calculator.proto"],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
@ -27,6 +27,18 @@ mediapipe_proto_library(
], ],
) )
mediapipe_cc_proto_library(
name = "iris_to_render_data_calculator_cc_proto",
srcs = ["iris_to_render_data_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
"//mediapipe/util:color_cc_proto",
"//mediapipe/util:render_data_cc_proto",
],
visibility = ["//visibility:public"],
deps = [":iris_to_render_data_calculator_proto"],
)
cc_library( cc_library(
name = "iris_to_render_data_calculator", name = "iris_to_render_data_calculator",
srcs = ["iris_to_render_data_calculator.cc"], srcs = ["iris_to_render_data_calculator.cc"],
@ -44,7 +56,7 @@ cc_library(
alwayslink = 1, alwayslink = 1,
) )
mediapipe_proto_library( proto_library(
name = "iris_to_depth_calculator_proto", name = "iris_to_depth_calculator_proto",
srcs = ["iris_to_depth_calculator.proto"], srcs = ["iris_to_depth_calculator.proto"],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
@ -53,6 +65,16 @@ mediapipe_proto_library(
], ],
) )
mediapipe_cc_proto_library(
name = "iris_to_depth_calculator_cc_proto",
srcs = ["iris_to_depth_calculator.proto"],
cc_deps = [
"//mediapipe/framework:calculator_cc_proto",
],
visibility = ["//visibility:public"],
deps = [":iris_to_depth_calculator_proto"],
)
cc_library( cc_library(
name = "iris_to_depth_calculator", name = "iris_to_depth_calculator",
srcs = ["iris_to_depth_calculator.cc"], srcs = ["iris_to_depth_calculator.cc"],

165
mediapipe/util/tracking/BUILD
View File

@ -13,24 +13,24 @@
# limitations under the License. # limitations under the License.
# #
load("//mediapipe/framework/port:build_config.bzl", "mediapipe_proto_library") load("//mediapipe/framework/port:build_config.bzl", "mediapipe_cc_proto_library")
licenses(["notice"]) licenses(["notice"])
package(default_visibility = ["//visibility:public"]) package(default_visibility = ["//visibility:public"])
mediapipe_proto_library( proto_library(
name = "tone_models_proto", name = "tone_models_proto",
srcs = ["tone_models.proto"], srcs = ["tone_models.proto"],
) )
mediapipe_proto_library( proto_library(
name = "tone_estimation_proto", name = "tone_estimation_proto",
srcs = ["tone_estimation.proto"], srcs = ["tone_estimation.proto"],
deps = [":tone_models_proto"], deps = [":tone_models_proto"],
) )
mediapipe_proto_library( proto_library(
name = "region_flow_computation_proto", name = "region_flow_computation_proto",
srcs = ["region_flow_computation.proto"], srcs = ["region_flow_computation.proto"],
deps = [ deps = [
@ -38,17 +38,17 @@ mediapipe_proto_library(
], ],
) )
mediapipe_proto_library( proto_library(
name = "motion_saliency_proto", name = "motion_saliency_proto",
srcs = ["motion_saliency.proto"], srcs = ["motion_saliency.proto"],
) )
mediapipe_proto_library( proto_library(
name = "motion_estimation_proto", name = "motion_estimation_proto",
srcs = ["motion_estimation.proto"], srcs = ["motion_estimation.proto"],
) )
mediapipe_proto_library( proto_library(
name = "motion_analysis_proto", name = "motion_analysis_proto",
srcs = ["motion_analysis.proto"], srcs = ["motion_analysis.proto"],
deps = [ deps = [
@ -58,33 +58,33 @@ mediapipe_proto_library(
], ],
) )
mediapipe_proto_library( proto_library(
name = "region_flow_proto", name = "region_flow_proto",
srcs = ["region_flow.proto"], srcs = ["region_flow.proto"],
) )
mediapipe_proto_library( proto_library(
name = "motion_models_proto", name = "motion_models_proto",
srcs = ["motion_models.proto"], srcs = ["motion_models.proto"],
) )
mediapipe_proto_library( proto_library(
name = "camera_motion_proto", name = "camera_motion_proto",
srcs = ["camera_motion.proto"], srcs = ["camera_motion.proto"],
deps = [":motion_models_proto"], deps = [":motion_models_proto"],
) )
mediapipe_proto_library( proto_library(
name = "push_pull_filtering_proto", name = "push_pull_filtering_proto",
srcs = ["push_pull_filtering.proto"], srcs = ["push_pull_filtering.proto"],
) )
mediapipe_proto_library( proto_library(
name = "frame_selection_solution_evaluator_proto", name = "frame_selection_solution_evaluator_proto",
srcs = ["frame_selection_solution_evaluator.proto"], srcs = ["frame_selection_solution_evaluator.proto"],
) )
mediapipe_proto_library( proto_library(
name = "frame_selection_proto", name = "frame_selection_proto",
srcs = ["frame_selection.proto"], srcs = ["frame_selection.proto"],
deps = [ deps = [
@ -94,7 +94,7 @@ mediapipe_proto_library(
], ],
) )
mediapipe_proto_library( proto_library(
name = "flow_packager_proto", name = "flow_packager_proto",
srcs = ["flow_packager.proto"], srcs = ["flow_packager.proto"],
deps = [ deps = [
@ -103,7 +103,7 @@ mediapipe_proto_library(
], ],
) )
mediapipe_proto_library( proto_library(
name = "tracking_proto", name = "tracking_proto",
srcs = ["tracking.proto"], srcs = ["tracking.proto"],
deps = [ deps = [
@ -111,18 +111,18 @@ mediapipe_proto_library(
], ],
) )
mediapipe_proto_library( proto_library(
name = "box_tracker_proto", name = "box_tracker_proto",
srcs = ["box_tracker.proto"], srcs = ["box_tracker.proto"],
deps = [":tracking_proto"], deps = [":tracking_proto"],
) )
mediapipe_proto_library( proto_library(
name = "tracked_detection_manager_config_proto", name = "tracked_detection_manager_config_proto",
srcs = ["tracked_detection_manager_config.proto"], srcs = ["tracked_detection_manager_config.proto"],
) )
mediapipe_proto_library( proto_library(
name = "box_detector_proto", name = "box_detector_proto",
srcs = ["box_detector.proto"], srcs = ["box_detector.proto"],
deps = [ deps = [
@ -131,6 +131,135 @@ mediapipe_proto_library(
], ],
) )
mediapipe_cc_proto_library(
name = "tone_models_cc_proto",
srcs = ["tone_models.proto"],
deps = [":tone_models_proto"],
)
mediapipe_cc_proto_library(
name = "tone_estimation_cc_proto",
srcs = ["tone_estimation.proto"],
cc_deps = [":tone_models_cc_proto"],
deps = [":tone_estimation_proto"],
)
mediapipe_cc_proto_library(
name = "region_flow_computation_cc_proto",
srcs = ["region_flow_computation.proto"],
cc_deps = [
":tone_estimation_cc_proto",
":tone_models_cc_proto",
],
deps = [":region_flow_computation_proto"],
)
mediapipe_cc_proto_library(
name = "motion_saliency_cc_proto",
srcs = ["motion_saliency.proto"],
deps = [":motion_saliency_proto"],
)
mediapipe_cc_proto_library(
name = "motion_estimation_cc_proto",
srcs = ["motion_estimation.proto"],
deps = [":motion_estimation_proto"],
)
mediapipe_cc_proto_library(
name = "motion_analysis_cc_proto",
srcs = ["motion_analysis.proto"],
cc_deps = [
":motion_estimation_cc_proto",
":motion_saliency_cc_proto",
":region_flow_computation_cc_proto",
],
deps = [":motion_analysis_proto"],
)
mediapipe_cc_proto_library(
name = "region_flow_cc_proto",
srcs = ["region_flow.proto"],
cc_deps = [":motion_models_cc_proto"],
deps = [":region_flow_proto"],
)
mediapipe_cc_proto_library(
name = "motion_models_cc_proto",
srcs = ["motion_models.proto"],
deps = [":motion_models_proto"],
)
mediapipe_cc_proto_library(
name = "camera_motion_cc_proto",
srcs = ["camera_motion.proto"],
cc_deps = [":motion_models_cc_proto"],
deps = [":camera_motion_proto"],
)
mediapipe_cc_proto_library(
name = "push_pull_filtering_cc_proto",
srcs = ["push_pull_filtering.proto"],
deps = [":push_pull_filtering_proto"],
)
mediapipe_cc_proto_library(
name = "frame_selection_solution_evaluator_cc_proto",
srcs = ["frame_selection_solution_evaluator.proto"],
deps = [":frame_selection_solution_evaluator_proto"],
)
mediapipe_cc_proto_library(
name = "frame_selection_cc_proto",
srcs = ["frame_selection.proto"],
cc_deps = [
":camera_motion_cc_proto",
":frame_selection_solution_evaluator_cc_proto",
":region_flow_cc_proto",
],
deps = [":frame_selection_proto"],
)
mediapipe_cc_proto_library(
name = "flow_packager_cc_proto",
srcs = ["flow_packager.proto"],
cc_deps = [
":motion_models_cc_proto",
":region_flow_cc_proto",
],
deps = [":flow_packager_proto"],
)
mediapipe_cc_proto_library(
name = "tracking_cc_proto",
srcs = ["tracking.proto"],
cc_deps = [":motion_models_cc_proto"],
deps = [":tracking_proto"],
)
mediapipe_cc_proto_library(
name = "box_tracker_cc_proto",
srcs = ["box_tracker.proto"],
cc_deps = [":tracking_cc_proto"],
deps = [":box_tracker_proto"],
)
mediapipe_cc_proto_library(
name = "tracked_detection_manager_config_cc_proto",
srcs = ["tracked_detection_manager_config.proto"],
deps = [":tracked_detection_manager_config_proto"],
)
mediapipe_cc_proto_library(
name = "box_detector_cc_proto",
srcs = ["box_detector.proto"],
cc_deps = [
":box_tracker_cc_proto",
":region_flow_cc_proto",
],
deps = [":box_detector_proto"],
)
cc_library( cc_library(
name = "motion_models", name = "motion_models",
srcs = ["motion_models.cc"], srcs = ["motion_models.cc"],