From 9045a74ba39e5760a86594d1ef06df859d0f788b Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Thu, 1 Jun 2023 17:45:03 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 537175065 --- mediapipe/calculators/core/BUILD | 1 + mediapipe/calculators/core/get_vector_item_calculator.cc | 8 ++++++++ 2 files changed, 9 insertions(+) diff --git a/mediapipe/calculators/core/BUILD b/mediapipe/calculators/core/BUILD index 297a416d..d773bef3 100644 --- a/mediapipe/calculators/core/BUILD +++ b/mediapipe/calculators/core/BUILD @@ -1240,6 +1240,7 @@ cc_library( "//mediapipe/framework/formats:classification_cc_proto", "//mediapipe/framework/formats:detection_cc_proto", "//mediapipe/framework/formats:landmark_cc_proto", + "//mediapipe/framework/formats:rect_cc_proto", "//mediapipe/framework/port:ret_check", "//mediapipe/framework/port:status", ], diff --git a/mediapipe/calculators/core/get_vector_item_calculator.cc b/mediapipe/calculators/core/get_vector_item_calculator.cc index 3306e4ff..5dbda6d9 100644 --- a/mediapipe/calculators/core/get_vector_item_calculator.cc +++ b/mediapipe/calculators/core/get_vector_item_calculator.cc @@ -17,6 +17,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/rect.pb.h" namespace mediapipe { namespace api2 { @@ -37,5 +38,12 @@ using GetDetectionVectorItemCalculator = GetVectorItemCalculator; REGISTER_CALCULATOR(GetDetectionVectorItemCalculator); +using GetNormalizedRectVectorItemCalculator = + GetVectorItemCalculator; +REGISTER_CALCULATOR(GetNormalizedRectVectorItemCalculator); + +using GetRectVectorItemCalculator = GetVectorItemCalculator; +REGISTER_CALCULATOR(GetRectVectorItemCalculator); + } // namespace api2 } // namespace mediapipe