diff --git a/mediapipe/tasks/cc/vision/object_detector/object_detector.cc b/mediapipe/tasks/cc/vision/object_detector/object_detector.cc index 5aecc2d2f..9149a3cbe 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 a2fb373cb..07e912cfc 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 {