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

View File

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