From 395d9d8ea21c93bbefb37ad980ad41f66b9a2f9f Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Sun, 27 Nov 2022 00:05:08 -0800
Subject: [PATCH] Instantiate GetDetectionVectorItemCalculator variant of
 GetVectorItemCalculator<>.

PiperOrigin-RevId: 491123314
---
 mediapipe/calculators/core/BUILD                         | 1 +
 mediapipe/calculators/core/get_vector_item_calculator.cc | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/mediapipe/calculators/core/BUILD b/mediapipe/calculators/core/BUILD
index 39837fadb..3b658eb5b 100644
--- a/mediapipe/calculators/core/BUILD
+++ b/mediapipe/calculators/core/BUILD
@@ -1299,6 +1299,7 @@ cc_library(
         "//mediapipe/framework/api2:packet",
         "//mediapipe/framework/api2:port",
         "//mediapipe/framework/formats:classification_cc_proto",
+        "//mediapipe/framework/formats:detection_cc_proto",
         "//mediapipe/framework/formats:landmark_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 51fb46b98..3306e4ff3 100644
--- a/mediapipe/calculators/core/get_vector_item_calculator.cc
+++ b/mediapipe/calculators/core/get_vector_item_calculator.cc
@@ -15,6 +15,7 @@
 #include "mediapipe/calculators/core/get_vector_item_calculator.h"
 
 #include "mediapipe/framework/formats/classification.pb.h"
+#include "mediapipe/framework/formats/detection.pb.h"
 #include "mediapipe/framework/formats/landmark.pb.h"
 
 namespace mediapipe {
@@ -32,5 +33,9 @@ using GetClassificationListVectorItemCalculator =
     GetVectorItemCalculator<mediapipe::ClassificationList>;
 REGISTER_CALCULATOR(GetClassificationListVectorItemCalculator);
 
+using GetDetectionVectorItemCalculator =
+    GetVectorItemCalculator<mediapipe::Detection>;
+REGISTER_CALCULATOR(GetDetectionVectorItemCalculator);
+
 }  // namespace api2
 }  // namespace mediapipe