diff --git a/data/image_train_item.py b/data/image_train_item.py index 2db8630..a79693f 100644 --- a/data/image_train_item.py +++ b/data/image_train_item.py @@ -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