230 lines
8.0 KiB
Python
230 lines
8.0 KiB
Python
# Copyright 2020 The MediaPipe Authors.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
load("@org_tensorflow//tensorflow:tensorflow.bzl", "pybind_extension")
|
|
|
|
licenses(["notice"]) # Apache 2.0
|
|
|
|
package(default_visibility = ["//visibility:public"])
|
|
|
|
pybind_extension(
|
|
name = "_framework_bindings",
|
|
srcs = ["framework_bindings.cc"],
|
|
linkopts = select({
|
|
"//third_party:opencv_source_build": [],
|
|
"//conditions:default": [
|
|
"-lopencv_core",
|
|
"-lopencv_imgproc",
|
|
"-lopencv_highgui",
|
|
"-lopencv_video",
|
|
"-lopencv_features2d",
|
|
"-lopencv_calib3d",
|
|
"-lopencv_imgcodecs",
|
|
],
|
|
}),
|
|
module_name = "_framework_bindings",
|
|
deps = [
|
|
":builtin_calculators",
|
|
":builtin_task_graphs",
|
|
"//mediapipe/python/pybind:calculator_graph",
|
|
"//mediapipe/python/pybind:image",
|
|
"//mediapipe/python/pybind:image_frame",
|
|
"//mediapipe/python/pybind:matrix",
|
|
"//mediapipe/python/pybind:packet",
|
|
"//mediapipe/python/pybind:packet_creator",
|
|
"//mediapipe/python/pybind:packet_getter",
|
|
"//mediapipe/python/pybind:resource_util",
|
|
"//mediapipe/python/pybind:timestamp",
|
|
"//mediapipe/python/pybind:validated_graph_config",
|
|
"//mediapipe/tasks/python/core/pybind:task_runner",
|
|
"@com_google_absl//absl/strings:str_format",
|
|
"@stblib//:stb_image",
|
|
# Type registration.
|
|
"//mediapipe/framework:basic_types_registration",
|
|
"//mediapipe/framework/formats:classification_registration",
|
|
"//mediapipe/framework/formats:detection_registration",
|
|
"//mediapipe/framework/formats:landmark_registration",
|
|
"//mediapipe/framework/formats:rect_registration",
|
|
"//mediapipe/modules/objectron/calculators:annotation_registration",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "builtin_calculators",
|
|
deps = [
|
|
"//mediapipe/calculators/core:gate_calculator",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/calculators/core:side_packet_to_stream_calculator",
|
|
"//mediapipe/calculators/core:split_proto_list_calculator",
|
|
"//mediapipe/calculators/core:string_to_int_calculator",
|
|
"//mediapipe/calculators/image:image_transformation_calculator",
|
|
"//mediapipe/calculators/util:detection_unique_id_calculator",
|
|
"//mediapipe/modules/face_detection:face_detection_full_range_cpu",
|
|
"//mediapipe/modules/face_detection:face_detection_short_range_cpu",
|
|
"//mediapipe/modules/face_landmark:face_landmark_front_cpu",
|
|
"//mediapipe/modules/hand_landmark:hand_landmark_tracking_cpu",
|
|
"//mediapipe/modules/holistic_landmark:holistic_landmark_cpu",
|
|
"//mediapipe/modules/objectron:objectron_cpu",
|
|
"//mediapipe/modules/palm_detection:palm_detection_cpu",
|
|
"//mediapipe/modules/pose_detection:pose_detection_cpu",
|
|
"//mediapipe/modules/pose_landmark:pose_landmark_by_roi_cpu",
|
|
"//mediapipe/modules/pose_landmark:pose_landmark_cpu",
|
|
"//mediapipe/modules/selfie_segmentation:selfie_segmentation_cpu",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "builtin_task_graphs",
|
|
deps = [
|
|
"//mediapipe/tasks/cc/vision/gesture_recognizer:gesture_recognizer_graph",
|
|
"//mediapipe/tasks/cc/vision/image_classifier:image_classifier_graph",
|
|
"//mediapipe/tasks/cc/vision/image_embedder:image_embedder_graph",
|
|
"//mediapipe/tasks/cc/vision/image_segmenter:image_segmenter_graph",
|
|
"//mediapipe/tasks/cc/vision/object_detector:object_detector_graph",
|
|
] + select({
|
|
# TODO: Build text_classifier_graph and text_embedder_graph on Windows.
|
|
# TODO: Build audio_classifier_graph and audio_embedder_graph on Windows.
|
|
"//mediapipe:windows": [],
|
|
"//conditions:default": [
|
|
"//mediapipe/tasks/cc/audio/audio_classifier:audio_classifier_graph",
|
|
"//mediapipe/tasks/cc/audio/audio_embedder:audio_embedder_graph",
|
|
"//mediapipe/tasks/cc/text/text_classifier:text_classifier_graph",
|
|
"//mediapipe/tasks/cc/text/text_embedder:text_embedder_graph",
|
|
],
|
|
}),
|
|
)
|
|
|
|
py_library(
|
|
name = "packet_creator",
|
|
srcs = ["packet_creator.py"],
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "packet_getter",
|
|
srcs = ["packet_getter.py"],
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "solution_base",
|
|
srcs = ["solution_base.py"],
|
|
srcs_version = "PY3",
|
|
visibility = [
|
|
"//mediapipe/python:__subpackages__",
|
|
],
|
|
deps = [
|
|
":_framework_bindings",
|
|
":packet_creator",
|
|
":packet_getter",
|
|
"//mediapipe/calculators/core:constant_side_packet_calculator_py_pb2",
|
|
"//mediapipe/calculators/image:image_transformation_calculator_py_pb2",
|
|
"//mediapipe/calculators/tensor:tensors_to_detections_calculator_py_pb2",
|
|
"//mediapipe/calculators/util:landmarks_smoothing_calculator_py_pb2",
|
|
"//mediapipe/calculators/util:logic_calculator_py_pb2",
|
|
"//mediapipe/calculators/util:thresholding_calculator_py_pb2",
|
|
"//mediapipe/framework:calculator_py_pb2",
|
|
"//mediapipe/framework/formats:classification_py_pb2",
|
|
"//mediapipe/framework/formats:detection_py_pb2",
|
|
"//mediapipe/framework/formats:landmark_py_pb2",
|
|
"//mediapipe/framework/formats:rect_py_pb2",
|
|
"//mediapipe/modules/objectron/calculators:annotation_py_pb2",
|
|
"//mediapipe/modules/objectron/calculators:lift_2d_frame_annotation_to_3d_calculator_py_pb2",
|
|
"@com_google_protobuf//:protobuf_python",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "calculator_graph_test",
|
|
srcs = ["calculator_graph_test.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
":packet_creator",
|
|
":packet_getter",
|
|
"//mediapipe/framework:calculator_py_pb2",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "image_test",
|
|
srcs = ["image_test.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "image_frame_test",
|
|
srcs = ["image_frame_test.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "packet_test",
|
|
srcs = ["packet_test.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
":packet_creator",
|
|
":packet_getter",
|
|
"//mediapipe/framework/formats:detection_py_pb2",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "timestamp_test",
|
|
srcs = ["timestamp_test.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":_framework_bindings",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "solution_base_test",
|
|
srcs = ["solution_base_test.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
":solution_base",
|
|
"//file/google_src",
|
|
"//file/localfile",
|
|
"//mediapipe/calculators/core:gate_calculator",
|
|
"//mediapipe/calculators/core:pass_through_calculator",
|
|
"//mediapipe/calculators/core:side_packet_to_stream_calculator",
|
|
"//mediapipe/calculators/image:image_transformation_calculator",
|
|
"//mediapipe/calculators/util:detection_unique_id_calculator",
|
|
"//mediapipe/calculators/util:to_image_calculator",
|
|
"//mediapipe/framework:calculator_py_pb2",
|
|
"//mediapipe/framework/formats:detection_py_pb2",
|
|
"@com_google_protobuf//:protobuf_python",
|
|
],
|
|
)
|