Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

removing the convert_rgb flag for a simpler design

This commit is contained in:
tyler 2023-04-14 20:21:48 -05:00
parent a839180199
commit b8f637873f
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
data/image_train_item.py
View File

@ -149,13 +149,10 @@ class ImageTrainItem:
self.error = None
self.__compute_target_width_height()
def load_image(self, convert_rgb=True):
def load_image(self):
try:
if convert_rgb:
image = PIL.Image.open(self.pathname).convert('RGB')
image = ImageOps.exif_transpose(image)
else:
image = PIL.Image.open(self.pathname)
image = PIL.Image.open(self.pathname).convert('RGB')
image = ImageOps.exif_transpose(image)
except SyntaxError as e:
pass
return image
@ -239,7 +236,7 @@ class ImageTrainItem:
def __compute_target_width_height(self):
self.target_wh = None
try:
with self.load_image(convert_rgb=False) as image:
with PIL.Image.open(self.pathname) as image:
width, height = image.size
image_aspect = width / height
target_wh = min(self.aspects, key=lambda aspects:abs(aspects[0]/aspects[1] - image_aspect))