From d42bf846aa4f3fca2f63bd51a66c54d8c98da0e3 Mon Sep 17 00:00:00 2001
From: Prianka Liz Kariat <prianka.kariat@ymedialabs.com>
Date: Tue, 12 Sep 2023 16:41:45 +0530
Subject: [PATCH] Updated multiple function in iOS Image Segmenter to use
 vector

---
 .../vision/image_segmenter/MPPImageSegmenterTests.mm | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm b/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
index 9df89ad2a..3ec02f365 100644
--- a/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
+++ b/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
@@ -44,14 +44,14 @@ double sum(const float *mask, size_t size) {
 
 float *multiply(const float *mask1, const float *mask2, size_t size) {
   double sum = 0.0;
-  float *multipliedMask = (float *)malloc(size * sizeof(float));
-  if (!multipliedMask) {
-    exit(-1);
-  }
+  
+  std::vector<float> multipliedMask;
+  multipliedMask.reserve(size);
+  
   for (int i = 0; i < size; i++) {
-    multipliedMask[i] = mask1[i] * mask2[i];
+    multipliedMask.push_back(mask1[i] * mask2[i]);
   }
-
+  
   return multipliedMask;
 }