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

Initialize target_wh to None

This commit is contained in:
Joel Holdbrooks 2023-01-24 08:33:40 -08:00
parent e6d1096637
commit a7505df0a6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
data/image_train_item.py
View File

@ -351,6 +351,7 @@ class ImageTrainItem:
return self return self
def __compute_target_width_height(self): def __compute_target_width_height(self):
self.target_wh = None
try: try:
with Image.open(self.pathname) as image: with Image.open(self.pathname) as image:
width, height = image.size width, height = image.size