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

Use SRGBA for Mac on Python for image test

Browse Source 
PiperOrigin-RevId: 577931014
This commit is contained in:
Sebastian Schmidt 2023-10-30 13:04:17 -07:00 committed by Copybara-Service
parent 7256bd2638
commit ec032fb018
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mediapipe/python/image_test.py
View File

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