| 1234567891011121314151617181920212223 |
- import os.path as osp
- import numpy as np
- import PIL.Image
- from labelme.utils import image as image_module
- from .util import data_dir
- from .util import get_img_and_data
- def test_img_b64_to_arr():
- img, _ = get_img_and_data()
- assert img.dtype == np.uint8
- assert img.shape == (907, 1210, 3)
- def test_img_arr_to_b64():
- img_file = osp.join(data_dir, 'apc2016_obj3.jpg')
- img_arr = np.asarray(PIL.Image.open(img_file))
- img_b64 = image_module.img_arr_to_b64(img_arr)
- img_arr2 = image_module.img_b64_to_arr(img_b64)
- np.testing.assert_allclose(img_arr, img_arr2)
|
