Skip to content

Commit

Permalink
removed prints
Browse files Browse the repository at this point in the history
  • Loading branch information
manole-alexandru committed Mar 10, 2023
1 parent a3da843 commit f4ab7eb
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions utils/dataloaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -507,7 +507,6 @@ def __init__(self,
nl = len(np.concatenate(labels, 0)) # number of labels
assert nl > 0 or not augment, f'{prefix}All labels empty in {cache_path}, can not start training. {HELP_URL}'
self.labels = list(labels)
print('Labels:', self.labels)
self.shapes = np.array(shapes)
self.im_files = list(cache.keys()) # update
self.label_files = img2label_paths(cache.keys()) # update
Expand Down Expand Up @@ -584,9 +583,7 @@ def __init__(self,
for i, x in pbar:
if cache_images == 'disk':
b += self.npy_files[i].stat().st_size
print('DISK')
else: # 'ram'
print('RAM')
self.ims[i], self.im_hw0[i], self.im_hw[i] = x # im, hw_orig, hw_resized = load_image(self, i)
b += self.ims[i].nbytes
pbar.desc = f'{prefix}Caching images ({b / gb:.1f}GB {cache_images})'
Expand Down

0 comments on commit f4ab7eb

Please sign in to comment.