diff --git a/mediapipe/calculators/core/BUILD b/mediapipe/calculators/core/BUILD index 8fbc9982..9c4b3af7 100644 --- a/mediapipe/calculators/core/BUILD +++ b/mediapipe/calculators/core/BUILD @@ -222,10 +222,10 @@ cc_library( "//mediapipe/framework:calculator_contract", "//mediapipe/framework:calculator_framework", "//mediapipe/framework:collection_item_id", - "//mediapipe/framework:packet", "//mediapipe/framework/formats:classification_cc_proto", "//mediapipe/framework/formats:detection_cc_proto", "//mediapipe/framework/formats:landmark_cc_proto", + "//mediapipe/framework/formats:matrix", "//mediapipe/framework/formats:rect_cc_proto", "//mediapipe/framework/port:integral_types", "//mediapipe/framework/port:ret_check", diff --git a/mediapipe/calculators/core/end_loop_calculator.cc b/mediapipe/calculators/core/end_loop_calculator.cc index b321f427..45cd8a9f 100644 --- a/mediapipe/calculators/core/end_loop_calculator.cc +++ b/mediapipe/calculators/core/end_loop_calculator.cc @@ -19,6 +19,7 @@ #include "mediapipe/framework/formats/classification.pb.h" #include "mediapipe/framework/formats/detection.pb.h" #include "mediapipe/framework/formats/landmark.pb.h" +#include "mediapipe/framework/formats/matrix.h" #include "mediapipe/framework/formats/rect.pb.h" #include "mediapipe/util/render_data.pb.h" #include "tensorflow/lite/interpreter.h" @@ -58,4 +59,7 @@ typedef EndLoopCalculator> EndLoopDetectionCalculator; REGISTER_CALCULATOR(EndLoopDetectionCalculator); +typedef EndLoopCalculator> EndLoopMatrixCalculator; +REGISTER_CALCULATOR(EndLoopMatrixCalculator); + } // namespace mediapipe