From 3015f5d1b6323d393c4f83b83200f4cbba21a976 Mon Sep 17 00:00:00 2001 From: ManoleAlexandru99 Date: Fri, 10 Mar 2023 18:09:00 +0200 Subject: [PATCH] Print #1 --- utils/dataloaders.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 12e0d964310b..8b6d46cb1bb0 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -509,6 +509,7 @@ def __init__(self, self.labels = list(labels) self.shapes = np.array(shapes) self.im_files = list(cache.keys()) # update + print('IMFILES', self.im_files[0]) self.label_files = img2label_paths(cache.keys()) # update # Filter images @@ -516,8 +517,6 @@ def __init__(self, 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] - print('IMFILES', self.im_files[0]) - print('INCLUDE', include) self.label_files = [self.label_files[i] for i in include] self.labels = [self.labels[i] for i in include] self.segments = [self.segments[i] for i in include] @@ -549,7 +548,6 @@ def __init__(self, s = self.shapes # wh ar = s[:, 1] / s[:, 0] # aspect ratio irect = ar.argsort() - print('IRECT:', irect) self.im_files = [self.im_files[i] for i in irect] self.label_files = [self.label_files[i] for i in irect] self.labels = [self.labels[i] for i in irect]