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

Fixed some typos in methods and comments

Browse Source
This commit is contained in:
kinaryml 2022-11-08 23:47:58 -08:00
parent dd6fdedd5f
commit d327b3649d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mediapipe/tasks/python/test/vision/image_embedder_test.py
View File

@ -167,7 +167,7 @@ class ImageEmbedderTest(parameterized.TestCase):
# Checks cosine similarity. # Checks cosine similarity.
self._check_cosine_similarity(image_result, crop_result, quantize, self._check_cosine_similarity(image_result, crop_result, quantize,
expected_similarity) expected_similarity)
# Closes the embedder explicitly when the classifier is not used in # Closes the embedder explicitly when the embedder is not used in
# a context. # a context.
embedder.close() embedder.close()
@ -315,7 +315,7 @@ class ImageEmbedderTest(parameterized.TestCase):
r'not initialized with the image mode'): r'not initialized with the image mode'):
embedder.embed(self.test_image) embedder.embed(self.test_image)
def test_calling_classify_for_video_in_live_stream_mode(self): def test_calling_embed_for_video_in_live_stream_mode(self):
options = _ImageEmbedderOptions( options = _ImageEmbedderOptions(
base_options=_BaseOptions(model_asset_path=self.model_path), base_options=_BaseOptions(model_asset_path=self.model_path),
running_mode=_RUNNING_MODE.LIVE_STREAM, running_mode=_RUNNING_MODE.LIVE_STREAM,
@ -325,7 +325,7 @@ class ImageEmbedderTest(parameterized.TestCase):
r'not initialized with the video mode'): r'not initialized with the video mode'):
embedder.embed_for_video(self.test_image, 0) embedder.embed_for_video(self.test_image, 0)
def test_classify_async_calls_with_illegal_timestamp(self): def test_embed_async_calls_with_illegal_timestamp(self):
options = _ImageEmbedderOptions( options = _ImageEmbedderOptions(
base_options=_BaseOptions(model_asset_path=self.model_path), base_options=_BaseOptions(model_asset_path=self.model_path),
running_mode=_RUNNING_MODE.LIVE_STREAM, running_mode=_RUNNING_MODE.LIVE_STREAM,
@ -365,7 +365,7 @@ class ImageEmbedderTest(parameterized.TestCase):
for timestamp in range(0, 300, 30): for timestamp in range(0, 300, 30):
embedder.embed_async(self.test_image, timestamp) embedder.embed_async(self.test_image, timestamp)
def test_classify_async_succeeds_with_region_of_interest(self): def test_embed_async_succeeds_with_region_of_interest(self):
# Get the embedding result for the cropped image. # Get the embedding result for the cropped image.
options = _ImageEmbedderOptions( options = _ImageEmbedderOptions(
base_options=_BaseOptions(model_asset_path=self.model_path), base_options=_BaseOptions(model_asset_path=self.model_path),