From 3c0ddf16b4c2b04cfff07d0db0aba48411468e9c Mon Sep 17 00:00:00 2001 From: Jiuqiang Tang Date: Wed, 7 Dec 2022 11:37:04 -0800 Subject: [PATCH] Fix an incorrect model sanity check in the object detector graph. PiperOrigin-RevId: 493663592 --- .../tasks/cc/vision/object_detector/object_detector_graph.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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 f5dc7e061..a1625c16c 100644 --- a/mediapipe/tasks/cc/vision/object_detector/object_detector_graph.cc +++ b/mediapipe/tasks/cc/vision/object_detector/object_detector_graph.cc @@ -532,8 +532,7 @@ class ObjectDetectorGraph : public core::ModelTaskGraph { MP_RETURN_IF_ERROR(SanityCheckOptions(task_options)); // Checks that the model has 4 outputs. auto& model = *model_resources.GetTfLiteModel(); - if (model.subgraphs()->size() != 1 || - (*model.subgraphs())[0]->outputs()->size() != 4) { + if (model.subgraphs()->size() != 1) { return CreateStatusWithPayload( absl::StatusCode::kInvalidArgument, absl::StrFormat("Expected a model with a single subgraph, found %d.",