Skip to content

GitLab

Unverified Commit b693aff7 authored by Christopher Friesen's avatar Christopher Friesen Committed by GitHub
Browse files

fix: resize transform now preserves aspect ratio (#1804)

parent 8a4c3e50
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/text_to_image/train_text_to_image.py
View file @ b693aff7
......@@ -493,7 +493,7 @@ def main():
train_transforms = transforms.Compose(
[
transforms.Resize((args.resolution, args.resolution), interpolation=transforms.InterpolationMode.BILINEAR),
transforms.Resize(args.resolution, interpolation=transforms.InterpolationMode.BILINEAR),
transforms.CenterCrop(args.resolution) if args.center_crop else transforms.RandomCrop(args.resolution),
transforms.RandomHorizontalFlip() if args.random_flip else transforms.Lambda(lambda x: x),
transforms.ToTensor(),
......
examples/text_to_image/train_text_to_image_flax.py
View file @ b693aff7
......@@ -333,7 +333,7 @@ def main():
train_transforms = transforms.Compose(
[
transforms.Resize((args.resolution, args.resolution), interpolation=transforms.InterpolationMode.BILINEAR),
transforms.Resize(args.resolution, interpolation=transforms.InterpolationMode.BILINEAR),
transforms.CenterCrop(args.resolution) if args.center_crop else transforms.RandomCrop(args.resolution),
transforms.RandomHorizontalFlip() if args.random_flip else transforms.Lambda(lambda x: x),
transforms.ToTensor(),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment