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
Hide 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(): ...@@ -493,7 +493,7 @@ def main():
train_transforms = transforms.Compose( 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.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.RandomHorizontalFlip() if args.random_flip else transforms.Lambda(lambda x: x),
transforms.ToTensor(), transforms.ToTensor(),
......
examples/text_to_image/train_text_to_image_flax.py
View file @ b693aff7
...@@ -333,7 +333,7 @@ def main(): ...@@ -333,7 +333,7 @@ def main():
train_transforms = transforms.Compose( 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.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.RandomHorizontalFlip() if args.random_flip else transforms.Lambda(lambda x: x),
transforms.ToTensor(), 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