RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Dataloader bug fixes

This commit is contained in:
Jaret Burkett 2023-08-27 14:36:38 -06:00
parent 9b164a8688
commit cc49786ee9
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/dataloader_mixins.py
View File

@ -107,6 +107,7 @@ class BucketsMixin:
# reduce it to the nearest divisible number
reduction = new_width % bucket_tolerance
file_item.crop_width = new_width - reduction
new_width = file_item.crop_width
# adjust the new x position so we evenly crop
file_item.crop_x = int(file_item.crop_x + (reduction / 2))
elif height > width:
@ -121,6 +122,7 @@ class BucketsMixin:
# reduce it to the nearest divisible number
reduction = new_height % bucket_tolerance
file_item.crop_height = new_height - reduction
new_height = file_item.crop_height
# adjust the new x position so we evenly crop
file_item.crop_y = int(file_item.crop_y + (reduction / 2))
else: