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

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

@ -44,12 +44,12 @@ double sum(const float *mask, size_t size) {
float *multiply(const float *mask1, const float *mask2, size_t size) { float *multiply(const float *mask1, const float *mask2, size_t size) {
double sum = 0.0; double sum = 0.0;
float *multipliedMask = (float *)malloc(size * sizeof(float));
if (!multipliedMask) { std::vector<float> multipliedMask;
exit(-1); multipliedMask.reserve(size);
}
for (int i = 0; i < size; i++) { for (int i = 0; i < size; i++) {
multipliedMask[i] = mask1[i] * mask2[i]; multipliedMask.push_back(mask1[i] * mask2[i]);
} }
return multipliedMask; return multipliedMask;