Add a new python unit test to test creating mediapipe Image from cvmat.
PiperOrigin-RevId: 495655719
This commit is contained in:
parent
299aa03302
commit
fd50b6aa2f
|
@ -28,6 +28,8 @@ import PIL.Image
|
||||||
from mediapipe.python._framework_bindings import image
|
from mediapipe.python._framework_bindings import image
|
||||||
from mediapipe.python._framework_bindings import image_frame
|
from mediapipe.python._framework_bindings import image_frame
|
||||||
|
|
||||||
|
TEST_IMAGE_PATH = 'mediapipe/python/solutions/testdata'
|
||||||
|
|
||||||
Image = image.Image
|
Image = image.Image
|
||||||
ImageFormat = image_frame.ImageFormat
|
ImageFormat = image_frame.ImageFormat
|
||||||
|
|
||||||
|
@ -187,5 +189,26 @@ 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_cvmat(self):
|
||||||
|
image_path = os.path.join(os.path.dirname(__file__),
|
||||||
|
'solutions/testdata/hands.jpg')
|
||||||
|
mat = cv2.imread(image_path).astype(np.uint8)
|
||||||
|
mat = cv2.cvtColor(mat, cv2.COLOR_BGR2RGB)
|
||||||
|
rgb_image = Image(image_format=ImageFormat.SRGB, data=mat)
|
||||||
|
self.assertEqual(rgb_image.width, 720)
|
||||||
|
self.assertEqual(rgb_image.height, 382)
|
||||||
|
self.assertEqual(rgb_image.channels, 3)
|
||||||
|
self.assertEqual(rgb_image.image_format, ImageFormat.SRGB)
|
||||||
|
self.assertTrue(np.array_equal(mat, rgb_image.numpy_view()))
|
||||||
|
|
||||||
|
def test_image_create_from_file(self):
|
||||||
|
image_path = os.path.join(os.path.dirname(__file__),
|
||||||
|
'solutions/testdata/hands.jpg')
|
||||||
|
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)
|
||||||
|
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