github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed issue with buckets scaling. again

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-08 16:32:14 -06:00
parent b01ab5d375
commit 4ed03a8d92
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
toolkit/dataloader_mixins.py
View File

@@ -100,6 +100,7 @@ class BucketsMixin:
# for file_item in enumerate(file_list): # for file_item in enumerate(file_list):
for idx, file_item in enumerate(file_list): for idx, file_item in enumerate(file_list):
file_item: 'FileItemDTO' = file_item
width = file_item.crop_width width = file_item.crop_width
height = file_item.crop_height height = file_item.crop_height
@@ -107,11 +108,11 @@ class BucketsMixin:
# set the scaling height and with to match smallest size, and keep aspect ratio # set the scaling height and with to match smallest size, and keep aspect ratio
if width > height: if width > height:
file_item.scale_height = bucket_resolution["height"] file_item.scale_to_height = bucket_resolution["height"]
file_item.scale_width = int(width * (bucket_resolution["height"] / height)) file_item.scale_to_width = int(width * (bucket_resolution["height"] / height))
else: else:
file_item.scale_width = bucket_resolution["width"] file_item.scale_to_width = bucket_resolution["width"]
file_item.scale_height = int(height * (bucket_resolution["width"] / width)) file_item.scale_to_height = int(height * (bucket_resolution["width"] / width))
file_item.crop_height = bucket_resolution["height"] file_item.crop_height = bucket_resolution["height"]
file_item.crop_width = bucket_resolution["width"] file_item.crop_width = bucket_resolution["width"]