"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "87c7864122e693948fee87c1a936c5b674c8a9b9"
Unverified Commit 4f2bf673 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Revert "Fix the order of width and height of original size in SDXL training script" (#5614)

Revert "Fix the order of width and height of original size in SDXL training script (#5382)"

This reverts commit 45db0499.
parent 29cf163b
...@@ -840,7 +840,7 @@ def main(args): ...@@ -840,7 +840,7 @@ def main(args):
all_images = [] all_images = []
crop_top_lefts = [] crop_top_lefts = []
for image in images: for image in images:
original_sizes.append((image.width, image.height)) original_sizes.append((image.height, image.width))
image = train_resize(image) image = train_resize(image)
if args.center_crop: if args.center_crop:
y1 = max(0, int(round((image.height - args.resolution) / 2.0))) y1 = max(0, int(round((image.height - args.resolution) / 2.0)))
......
...@@ -825,7 +825,7 @@ def main(args): ...@@ -825,7 +825,7 @@ def main(args):
all_images = [] all_images = []
crop_top_lefts = [] crop_top_lefts = []
for image in images: for image in images:
original_sizes.append((image.width, image.height)) original_sizes.append((image.height, image.width))
image = train_resize(image) image = train_resize(image)
if args.center_crop: if args.center_crop:
y1 = max(0, int(round((image.height - args.resolution) / 2.0))) y1 = max(0, int(round((image.height - args.resolution) / 2.0)))
......
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