def time_load(self, shape, mode): if mode == 'pillow': try: img = np.asarray( Image.open(self.filename_pillow).convert('RGBA')) except Image.DecompressionBombError: pass elif mode == 'imageio': img = imread(self.filename) else: img = imread(self.filename) assert np.array_equal(img, self.img)
def test_uint8_rgb_image(tmpdir, shape, backend): tmpfile = os.path.join(str(tmpdir), 'test.bmp') img = np.random.randint(255, size=shape, dtype=np.uint8) imwrite(tmpfile, img) if backend == 'pillow': img_read = np.asarray(Image.open(tmpfile).convert('RGB')) else: img_read = imread(tmpfile) assert img.dtype == img_read.dtype assert_array_equal(img, img_read)
def test_bool_image(tmpdir, shape, backend): tmpfile = os.path.join(str(tmpdir), 'test.bmp') img = np.random.randint(2, size=shape, dtype=np.bool) img[0, 0] = False img[-1, -1] = True print(img) imwrite(tmpfile, img) if backend == 'pillow': img_read = np.asarray(Image.open(tmpfile).convert('L')) else: img_read = imread(tmpfile) print(img_read) assert img_read[0, 0] == 0 assert img_read[-1, -1] == 255 color_pallet = np.asarray([0, 255], dtype=np.uint8) assert_array_equal(color_pallet[img.astype(np.uint8)], img_read)