From 59e0b1ba7454b57599fde07c0fe7d145b3366887 Mon Sep 17 00:00:00 2001 From: Prianka Liz Kariat Date: Thu, 23 Mar 2023 19:51:25 +0530 Subject: [PATCH] Added stream info for some modes in MPPImageClassifier --- .../tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.h | 1 + .../tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.mm | 5 +++++ .../utils/sources/MPPImageClassifierOptions+Helpers.mm | 5 +++-- 3 files changed, 9 insertions(+), 2 deletions(-) diff --git a/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.h b/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.h index d52df2ae4..d1810d56b 100644 --- a/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.h +++ b/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.h @@ -20,6 +20,7 @@ NS_ASSUME_NONNULL_BEGIN @interface MPPBaseOptions (Helpers) - (void)copyToProto:(mediapipe::tasks::core::proto::BaseOptions *)baseOptionsProto; +- (void)copyToProto:(mediapipe::tasks::core::proto::BaseOptions *)baseOptionsProto withUseStreamMode:(BOOL)useStreamMode; @end diff --git a/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.mm b/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.mm index 42cafe610..f7f8e5a55 100644 --- a/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.mm +++ b/mediapipe/tasks/ios/core/utils/sources/MPPBaseOptions+Helpers.mm @@ -22,6 +22,11 @@ using BaseOptionsProto = ::mediapipe::tasks::core::proto::BaseOptions; @implementation MPPBaseOptions (Helpers) +- (void)copyToProto:(BaseOptionsProto *)baseOptionsProto withUseStreamMode:(BOOL)useStreamMode { + [self copyToProto:baseOptionsProto]; + baseOptionsProto->set_use_stream_mode(useStreamMode); +} + - (void)copyToProto:(BaseOptionsProto *)baseOptionsProto { baseOptionsProto->Clear(); diff --git a/mediapipe/tasks/ios/vision/image_classifier/utils/sources/MPPImageClassifierOptions+Helpers.mm b/mediapipe/tasks/ios/vision/image_classifier/utils/sources/MPPImageClassifierOptions+Helpers.mm index 36ecf9093..926099c1a 100644 --- a/mediapipe/tasks/ios/vision/image_classifier/utils/sources/MPPImageClassifierOptions+Helpers.mm +++ b/mediapipe/tasks/ios/vision/image_classifier/utils/sources/MPPImageClassifierOptions+Helpers.mm @@ -32,8 +32,9 @@ using ClassifierOptionsProto = ::mediapipe::tasks::components::processors::proto - (void)copyToProto:(CalculatorOptionsProto *)optionsProto { ImageClassifierGraphOptionsProto *graphOptions = optionsProto->MutableExtension(ImageClassifierGraphOptionsProto::ext); - [self.baseOptions copyToProto:graphOptions->mutable_base_options()]; - + + [self.baseOptions copyToProto:graphOptions->mutable_base_options() withUseStreamMode:self.runningMode != MPPRunningModeImage]; + ClassifierOptionsProto *classifierOptionsProto = graphOptions->mutable_classifier_options(); classifierOptionsProto->Clear();