From 14dd4241f09d674b9a0331b1e04d6e9be697eded Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Mon, 26 Feb 2024 07:50:59 -0500 Subject: [PATCH] resize first --- examples/textual_inversion/textual_inversion_sdxl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/textual_inversion/textual_inversion_sdxl.py b/examples/textual_inversion/textual_inversion_sdxl.py index 90f1b36b08ea..d35a11b70a20 100644 --- a/examples/textual_inversion/textual_inversion_sdxl.py +++ b/examples/textual_inversion/textual_inversion_sdxl.py @@ -546,6 +546,8 @@ def __getitem__(self, i): example["original_size"] = (image.height, image.width) + image = image.resize((self.size, self.size), resample=self.interpolation) + if self.center_crop: y1 = max(0, int(round((image.height - self.size) / 2.0))) x1 = max(0, int(round((image.width - self.size) / 2.0))) @@ -576,7 +578,6 @@ def __getitem__(self, i): img = np.array(image).astype(np.uint8) image = Image.fromarray(img) - image = image.resize((self.size, self.size), resample=self.interpolation) image = self.flip_transform(image) image = np.array(image).astype(np.uint8)