Internal changes
PiperOrigin-RevId: 477924417
This commit is contained in:
parent
3a3a470a0c
commit
3225372c28
|
@ -187,16 +187,5 @@ class ImageTest(absltest.TestCase):
|
||||||
gc.collect()
|
gc.collect()
|
||||||
self.assertEqual(sys.getrefcount(rgb_image), initial_ref_count)
|
self.assertEqual(sys.getrefcount(rgb_image), initial_ref_count)
|
||||||
|
|
||||||
def test_image_create_from_file(self):
|
|
||||||
image_path = os.path.join(
|
|
||||||
resources.GetRunfilesDir(),
|
|
||||||
'mediapipe/tasks/testdata/vision/cat.jpg')
|
|
||||||
loaded_image = Image.create_from_file(image_path)
|
|
||||||
self.assertEqual(loaded_image.width, 600)
|
|
||||||
self.assertEqual(loaded_image.height, 400)
|
|
||||||
self.assertEqual(loaded_image.channels, 3)
|
|
||||||
self.assertEqual(loaded_image.image_format, ImageFormat.SRGB)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
absltest.main()
|
absltest.main()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user