From e6d10966376585579700e533377a9ffc1e7cb55d Mon Sep 17 00:00:00 2001 From: Joel Holdbrooks Date: Tue, 24 Jan 2023 08:33:20 -0800 Subject: [PATCH] Filter out items with errors --- data/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/data_loader.py b/data/data_loader.py index 8ea2d4b..2cff523 100644 --- a/data/data_loader.py +++ b/data/data_loader.py @@ -166,7 +166,7 @@ class DataLoaderMultiAspect(): print (f" * DLMA: {len(items)} images loaded from {len(image_paths)} files") - self.prepared_train_data = items + self.prepared_train_data = [item for item in items if item.error is None] random.Random(self.seed).shuffle(self.prepared_train_data) self.__report_errors(items)