talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplified async test cases to invoke the classifier in context

Browse Source
This commit is contained in:
kinaryml
2022-10-14 03:00:29 -07:00
parent 6771fe69e9
commit 803210a86b
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/python/test/vision/image_classifier_test.py
+6 -8
View File
@@ -471,10 +471,9 @@ class ImageClassifierTest(parameterized.TestCase):
running_mode=_RUNNING_MODE.LIVE_STREAM, running_mode=_RUNNING_MODE.LIVE_STREAM,
classifier_options=classifier_options, classifier_options=classifier_options,
result_callback=check_result) result_callback=check_result)
classifier = _ImageClassifier.create_from_options(options) with _ImageClassifier.create_from_options(options) as classifier:
for timestamp in range(0, 300, 30): for timestamp in range(0, 300, 30):
classifier.classify_async(self.test_image, timestamp) classifier.classify_async(self.test_image, timestamp)
classifier.close()
def test_classify_async_succeeds_with_region_of_interest(self): def test_classify_async_succeeds_with_region_of_interest(self):
# Load the test image. # Load the test image.
@@ -499,10 +498,9 @@ class ImageClassifierTest(parameterized.TestCase):
running_mode=_RUNNING_MODE.LIVE_STREAM, running_mode=_RUNNING_MODE.LIVE_STREAM,
classifier_options=classifier_options, classifier_options=classifier_options,
result_callback=check_result) result_callback=check_result)
classifier = _ImageClassifier.create_from_options(options) with _ImageClassifier.create_from_options(options) as classifier:
for timestamp in range(0, 300, 30): for timestamp in range(0, 300, 30):
classifier.classify_async(test_image, timestamp, roi) classifier.classify_async(test_image, timestamp, roi)
classifier.close()
if __name__ == '__main__': if __name__ == '__main__':