Skip to content

Commit b0290d6

Browse files
committed
Added checking for score and label of first category for tests of classify
1 parent 418659a commit b0290d6

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tensorflow_lite_support/c/test/task/vision/image_classifier_test.cc

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,10 @@ TEST_F(ImageClassifierClassifyTest, SucceedsWithImageData) {
212212
EXPECT_NE(classification_result->classifications, nullptr);
213213
EXPECT_GE(classification_result->classifications->size, 1);
214214
EXPECT_NE(classification_result->classifications->categories, nullptr);
215-
// TODO(prianka): check score and labels`
215+
EXPECT_EQ(strcmp(classification_result->classifications->categories[0].label,
216+
"cheeseburger"),
217+
0);
218+
EXPECT_GE(classification_result->classifications->categories[0].score, 0.90);
216219

217220
TfLiteClassificationResultDelete(classification_result);
218221
}
@@ -287,7 +290,10 @@ TEST_F(ImageClassifierClassifyTest, SucceedsWithRoiWithinImageBounds) {
287290
EXPECT_NE(classification_result->classifications, nullptr);
288291
EXPECT_GE(classification_result->classifications->size, 1);
289292
EXPECT_NE(classification_result->classifications->categories, nullptr);
290-
// TODO(prianka): check score and labels`
293+
EXPECT_EQ(strcmp(classification_result->classifications->categories[0].label,
294+
"bagel"),
295+
0);
296+
EXPECT_GE(classification_result->classifications->categories[0].score, 0.30);
291297

292298
TfLiteClassificationResultDelete(classification_result);
293299
}

0 commit comments

Comments
 (0)