def make_dataset(root, image_size=256):
transform = transforms.ComposedTransform(
transforms.Resize(image_size),
transforms.CenterCrop(image_size),
OptionalGrayscaleToFakeGrayscale(),
)
return ImageFolderDataset(root, transform=transform)
def test_Resize_image_size(self):
def PILResizeTransform(image_size):
size = image_size[::-1]
return lambda image: image.resize(size, resample=Image.BILINEAR)
image_size = (100, 100)
pystiche_transform = transforms.Resize(image_size)
pil_transform = PILResizeTransform(image_size)
self.assertTransformEqualsPIL(
pystiche_transform=pystiche_transform,
pil_transform=pil_transform,
tolerance=3e-2,
)
def test_Resize_image_size():
def PILResizeTransform(image_size):
size = image_size[::-1]
return lambda image: image.resize(size, resample=Image.BILINEAR)
image_size = (100, 100)
pystiche_transform = transforms.Resize(image_size)
pil_transform = PILResizeTransform(image_size)
assert_transform_equals_pil(
pystiche_transform=pystiche_transform,
pil_transform=pil_transform,
mae=3e-2,
)
def test_Resize_edge_size(self):
def PILFixedAspectRatioResizeTransform(edge_size, edge):
def transform(image):
aspect_ratio = calculate_aspect_ratio(image.size[::-1])
image_size = edge_to_image_size(edge_size, aspect_ratio, edge)
size = image_size[::-1]
return image.resize(size, resample=Image.BILINEAR)
return transform
edge_size = 100
for edge in ("short", "long", "vert", "horz"):
pystiche_transform = transforms.Resize(edge_size, edge=edge)
pil_transform = PILFixedAspectRatioResizeTransform(edge_size,
edge=edge)
self.assertTransformEqualsPIL(
pystiche_transform=pystiche_transform,
pil_transform=pil_transform,
tolerance=3e-2,
)