cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated multiple function in iOS Image Segmenter to use vector

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-09-12 16:41:45 +05:30
parent 15c8a25dc3
commit d42bf846aa
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
View File

@ -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;
}