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

No public description

Browse Source 
PiperOrigin-RevId: 577873616
This commit is contained in:
Sebastian Schmidt 2023-10-30 09:44:11 -07:00 committed by Copybara-Service
parent 2f4d7b4079
commit 7256bd2638
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/python/image_test.py
View File

@ -207,7 +207,9 @@ class ImageTest(absltest.TestCase):
loaded_image = Image.create_from_file(image_path)
self.assertEqual(loaded_image.width, 720)
self.assertEqual(loaded_image.height, 382)
self.assertEqual(loaded_image.channels, 3)
# On Mac w/ GPU support, images use 4 channels. Otherwise, all images use
# 3 channels.
self.assertIn(loaded_image.channels, [3, 4])
self.assertEqual(loaded_image.image_format, ImageFormat.SRGB)
if __name__ == '__main__':