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

Updated boolean test in iOS image segmenter tests

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-09-11 18:46:19 +05:30
parent 0ec58f4749
commit e5df5adf84
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -153,8 +153,13 @@ double softIOU(const float *mask1, const float *mask2, size_t size) {
usingImageSegmenter:imageSegmenter]; usingImageSegmenter:imageSegmenter];
XCTAssertNotNil(result.categoryMask); XCTAssertNotNil(result.categoryMask);
shouldHaveConfidenceMasks ? XCTAssertNotNil(result.confidenceMasks)
: XCTAssertNil(result.confidenceMasks); if (shouldHaveConfidenceMasks) {
XCTAssertNotNil(result.confidenceMasks);
}
else {
XCTAssertNil(result.confidenceMasks);
}
[self assertCategoryMask:result.categoryMask [self assertCategoryMask:result.categoryMask
approximatelyEqualsExpectedCategoryMaskImageWithFileInfo:expectedCategoryMaskFileInfo]; approximatelyEqualsExpectedCategoryMaskImageWithFileInfo:expectedCategoryMaskFileInfo];
@ -170,7 +175,13 @@ double softIOU(const float *mask1, const float *mask2, size_t size) {
usingImageSegmenter:imageSegmenter]; usingImageSegmenter:imageSegmenter];
XCTAssertNotNil(result.confidenceMasks); XCTAssertNotNil(result.confidenceMasks);
shouldHaveCategoryMask ? XCTAssertNotNil(result.categoryMask) : XCTAssertNil(result.categoryMask);
if (shouldHaveCategoryMask) {
XCTAssertNotNil(result.categoryMask);
}
else {
XCTAssertNil(result.categoryMask);
}
XCTAssertLessThan(index, result.confidenceMasks.count); XCTAssertLessThan(index, result.confidenceMasks.count);