Skip to content

Commit

Permalink
Add tqdm pbar.close() (ultralytics#2644)
Browse files Browse the repository at this point in the history
When using tqdm, sometimes it can't print in one line and roll to next line.

(cherry picked from commit 1e8ab3f)
  • Loading branch information
zzttqu authored and Lechtr committed Apr 1, 2021
1 parent 7877ac8 commit 697472e
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions utils/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -443,7 +443,8 @@ def __init__(self, path, img_size=640, batch_size=16, augment=False, hyp=None, r
self.imgs[i], self.img_hw0[i], self.img_hw[i] = x # img, hw_original, hw_resized = load_image(self, i)
gb += self.imgs[i].nbytes
pbar.desc = f'{prefix}Caching images ({gb / 1E9:.1f}GB)'

pbar.close()

def cache_labels(self, path=Path('./labels.cache'), prefix=''):
# Cache dataset labels, check images and read shapes
x = {} # dict
Expand Down Expand Up @@ -487,7 +488,8 @@ def cache_labels(self, path=Path('./labels.cache'), prefix=''):

pbar.desc = f"{prefix}Scanning '{path.parent / path.stem}' images and labels... " \
f"{nf} found, {nm} missing, {ne} empty, {nc} corrupted"

pbar.close()

if nf == 0:
print(f'{prefix}WARNING: No labels found in {path}. See {help_url}')

Expand Down

0 comments on commit 697472e

Please sign in to comment.