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