From a7505df0a69fe8ffbb5c593c3fab409bcf879a25 Mon Sep 17 00:00:00 2001 From: Joel Holdbrooks Date: Tue, 24 Jan 2023 08:33:40 -0800 Subject: [PATCH] Initialize target_wh to None --- data/image_train_item.py | 1 + 1 file changed, 1 insertion(+) 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