From fecda42344ea59d17614107edfa45d3652d5bf33 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 1 Nov 2022 16:33:51 +0100 Subject: [PATCH] Fix `min_items` Signed-off-by: Glenn Jocher --- utils/dataloaders.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 6b6e83e30456..4e5b75edb5c2 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -509,8 +509,8 @@ def __init__(self, # Filter images if min_items: - include = np.array([len(x) > min_items for x in self.labels]).nonzero()[0].astype(int) - LOGGER.info(f'{prefix}{nf - len(include)}/{nf} images filtered from dataset') + include = np.array([len(x) >= min_items for x in self.labels]).nonzero()[0].astype(int) + LOGGER.info(f'{prefix}{n - len(include)}/{n} images filtered from dataset') self.im_files = [self.im_files[i] for i in include] self.label_files = [self.label_files[i] for i in include] self.labels = [self.labels[i] for i in include]