def test_errors(self):
with self.assertRaises(TypeError):
F.to_tensor(1)
with self.assertRaises(ValueError):
fake_img = Image.fromarray((np.random.rand(28, 28, 3) * 255).astype(
'uint8'))
F.to_tensor(fake_img, data_format=1)
with self.assertRaises(ValueError):
fake_img = paddle.rand((3, 100, 100))
F.pad(fake_img, 1, padding_mode='symmetric')
with self.assertRaises(TypeError):
fake_img = paddle.rand((3, 100, 100))
F.resize(fake_img, {1: 1})
with self.assertRaises(TypeError):
fake_img = Image.fromarray((np.random.rand(28, 28, 3) * 255).astype(
'uint8'))
F.resize(fake_img, '1')
with self.assertRaises(TypeError):
F.resize(1, 1)
with self.assertRaises(TypeError):
F.pad(1, 1)
with self.assertRaises(TypeError):
F.crop(1, 1, 1, 1, 1)
with self.assertRaises(TypeError):
F.hflip(1)
with self.assertRaises(TypeError):
F.vflip(1)
with self.assertRaises(TypeError):
F.adjust_brightness(1, 0.1)
with self.assertRaises(TypeError):
F.adjust_contrast(1, 0.1)
with self.assertRaises(TypeError):
F.adjust_hue(1, 0.1)
with self.assertRaises(TypeError):
F.adjust_saturation(1, 0.1)
with self.assertRaises(TypeError):
F.rotate(1, 0.1)
with self.assertRaises(TypeError):
F.to_grayscale(1)
with self.assertRaises(ValueError):
set_image_backend(1)
with self.assertRaises(ValueError):
image_load('tmp.jpg', backend=1)
def test_rotate1(self):
np_img = (np.random.rand(28, 28, 3) * 255).astype('uint8')
pil_img = Image.fromarray(np_img).convert('RGB')
rotated_np_img = F.rotate(
np_img, 80, expand=True, center=[0, 0], fill=[0, 0, 0])
rotated_pil_img = F.rotate(
pil_img, 80, expand=True, center=[0, 0], fill=[0, 0, 0])
np.testing.assert_equal(rotated_np_img.shape,
np.array(rotated_pil_img).shape)
def _apply_image(self, img):
"""
Args:
img (PIL.Image|np.array): Image to be rotated.
Returns:
PIL.Image or np.array: Rotated image.
"""
angle = self.params
return F.rotate(img, angle, self.resample, self.expand, self.center,
self.fill)
def test_rotate(self):
np_img = (np.random.rand(28, 28, 3) * 255).astype('uint8')
pil_img = Image.fromarray(np_img).convert('RGB')
rotated_np_img = F.rotate(np_img, 80, expand=True)
rotated_pil_img = F.rotate(pil_img, 80, expand=True)
tensor_img = F.to_tensor(pil_img, 'CHW')
rotated_tensor_img1 = F.rotate(tensor_img, 80, expand=True)
rotated_tensor_img2 = F.rotate(tensor_img,
80,
interpolation='bilinear',
center=(10, 10),
expand=False)
np.testing.assert_equal(rotated_np_img.shape,
np.array(rotated_pil_img).shape)
np.testing.assert_equal(rotated_np_img.shape,
rotated_tensor_img1.transpose((1, 2, 0)).shape)
def test_errors(self):
with self.assertRaises(TypeError):
F.to_tensor(1)
with self.assertRaises(ValueError):
fake_img = Image.fromarray(
(np.random.rand(28, 28, 3) * 255).astype('uint8'))
F.to_tensor(fake_img, data_format=1)
with self.assertRaises(ValueError):
fake_img = paddle.rand((3, 100, 100))
F.pad(fake_img, 1, padding_mode='symmetric')
with self.assertRaises(TypeError):
fake_img = paddle.rand((3, 100, 100))
F.resize(fake_img, {1: 1})
with self.assertRaises(TypeError):
fake_img = Image.fromarray(
(np.random.rand(28, 28, 3) * 255).astype('uint8'))
F.resize(fake_img, '1')
with self.assertRaises(TypeError):
F.resize(1, 1)
with self.assertRaises(TypeError):
F.pad(1, 1)
with self.assertRaises(TypeError):
F.crop(1, 1, 1, 1, 1)
with self.assertRaises(TypeError):
F.hflip(1)
with self.assertRaises(TypeError):
F.vflip(1)
with self.assertRaises(TypeError):
F.adjust_brightness(1, 0.1)
with self.assertRaises(TypeError):
F.adjust_contrast(1, 0.1)
with self.assertRaises(TypeError):
F.adjust_hue(1, 0.1)
with self.assertRaises(TypeError):
F.adjust_saturation(1, 0.1)
with self.assertRaises(TypeError):
F.affine('45')
with self.assertRaises(TypeError):
F.affine(45, translate=0.3)
with self.assertRaises(TypeError):
F.affine(45, translate=[0.2, 0.2, 0.3])
with self.assertRaises(TypeError):
F.affine(45, translate=[0.2, 0.2], scale=-0.5)
with self.assertRaises(TypeError):
F.affine(45, translate=[0.2, 0.2], scale=0.5, shear=10)
with self.assertRaises(TypeError):
F.affine(45, translate=[0.2, 0.2], scale=0.5, shear=[-10, 0, 10])
with self.assertRaises(TypeError):
F.affine(45,
translate=[0.2, 0.2],
scale=0.5,
shear=[-10, 10],
interpolation=2)
with self.assertRaises(TypeError):
F.affine(45,
translate=[0.2, 0.2],
scale=0.5,
shear=[-10, 10],
center=0)
with self.assertRaises(TypeError):
F.rotate(1, 0.1)
with self.assertRaises(TypeError):
F.to_grayscale(1)
with self.assertRaises(ValueError):
set_image_backend(1)
with self.assertRaises(ValueError):
image_load('tmp.jpg', backend=1)
def _apply_mask(self, mask):
return F.rotate(mask, self.degrees, self.resample, self.expand,
self.center, self.fill)
def _apply_image(self, image):
return F.rotate(image, self.degrees, self.resample, self.expand,
self.center, self.fill)