zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 12:49:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix image cropping

Browse Source
This commit is contained in:
Angelos Katharopoulos
2024-10-10 03:17:23 -07:00
parent fc88e3b0d0
commit 8c3b25f88c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
flux/dreambooth.py
View File

@@ -53,10 +53,12 @@ class FinetuningDataset:
if width != height:
side = min(width, height)
img = img.crop(
(width - side) / 2,
(height - side) / 2,
(width + side) / 2,
(height + side) / 2,
(
(width - side) / 2,
(height - side) / 2,
(width + side) / 2,
(height + side) / 2,
)
)
img = img.resize(self.args.resolution, Image.LANCZOS)
img = mx.array(np.array(img))