From ba93bab286e170ca13b78cd68d10db42d480a8a2 Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Mon, 17 Oct 2022 16:56:12 -0700
Subject: [PATCH] Fix ObjectDetector C++ flow limiter and improve
 documentation.

PiperOrigin-RevId: 481774191
---
 mediapipe/tasks/cc/vision/object_detector/object_detector.cc  | 4 ++--
 .../tasks/cc/vision/object_detector/object_detector_graph.cc  | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/mediapipe/tasks/cc/vision/object_detector/object_detector.cc b/mediapipe/tasks/cc/vision/object_detector/object_detector.cc
index 5aecc2d2..9149a3cb 100644
--- a/mediapipe/tasks/cc/vision/object_detector/object_detector.cc
+++ b/mediapipe/tasks/cc/vision/object_detector/object_detector.cc
@@ -101,8 +101,8 @@ CalculatorGraphConfig CreateGraphConfig(
   task_subgraph.Out(kImageTag).SetName(kImageOutStreamName) >>
       graph.Out(kImageTag);
   if (enable_flow_limiting) {
-    return tasks::core::AddFlowLimiterCalculator(graph, task_subgraph,
-                                                 {kImageTag}, kDetectionsTag);
+    return tasks::core::AddFlowLimiterCalculator(
+        graph, task_subgraph, {kImageTag, kNormRectTag}, kDetectionsTag);
   }
   graph.In(kImageTag) >> task_subgraph.In(kImageTag);
   graph.In(kNormRectTag) >> task_subgraph.In(kNormRectTag);
diff --git a/mediapipe/tasks/cc/vision/object_detector/object_detector_graph.cc b/mediapipe/tasks/cc/vision/object_detector/object_detector_graph.cc
index a2fb373c..07e912cf 100644
--- a/mediapipe/tasks/cc/vision/object_detector/object_detector_graph.cc
+++ b/mediapipe/tasks/cc/vision/object_detector/object_detector_graph.cc
@@ -460,7 +460,7 @@ void ConfigureTensorsToDetectionsCalculator(
 //   IMAGE - Image
 //     Image to perform detection on.
 //   NORM_RECT - NormalizedRect @Optional
-//     Describes image rotation and region of image to perform classification
+//     Describes image rotation and region of image to perform detection
 //     on.
 //     @Optional: rect covering the whole image is used if not specified.
 //
@@ -469,6 +469,8 @@ void ConfigureTensorsToDetectionsCalculator(
 //     Detected objects with bounding box in pixel units.
 //   IMAGE - mediapipe::Image
 //     The image that object detection runs on.
+// All returned coordinates are in the unrotated and uncropped input image
+// coordinates system.
 //
 // Example:
 // node {