Skip to content

Commit

Permalink
Console corrupted -> corrupt (#6338)
Browse files Browse the repository at this point in the history
* Console corrupted -> corrupt 

Minor style changes.

* Update export.py
  • Loading branch information
glenn-jocher authored Jan 18, 2022
1 parent 3119b2f commit fd55271
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 5 deletions.
3 changes: 1 addition & 2 deletions export.py
Original file line number Diff line number Diff line change
Expand Up @@ -452,8 +452,7 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path'
assert not (tflite and tfjs), 'TFLite and TF.js models must be exported separately, please pass only one type.'
model, f = export_saved_model(model, im, file, dynamic, tf_nms=nms or agnostic_nms or tfjs,
agnostic_nms=agnostic_nms or tfjs, topk_per_class=topk_per_class,
topk_all=topk_all,
conf_thres=conf_thres, iou_thres=iou_thres) # keras model
topk_all=topk_all, conf_thres=conf_thres, iou_thres=iou_thres) # keras model
if pb or tfjs: # pb prerequisite to tfjs
f = export_pb(model, im, file)
if tflite or edgetpu:
Expand Down
6 changes: 3 additions & 3 deletions utils/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -424,9 +424,9 @@ def __init__(self, path, img_size=640, batch_size=16, augment=False, hyp=None, r
cache, exists = self.cache_labels(cache_path, prefix), False # cache

# Display cache
nf, nm, ne, nc, n = cache.pop('results') # found, missing, empty, corrupted, total
nf, nm, ne, nc, n = cache.pop('results') # found, missing, empty, corrupt, total
if exists:
d = f"Scanning '{cache_path}' images and labels... {nf} found, {nm} missing, {ne} empty, {nc} corrupted"
d = f"Scanning '{cache_path}' images and labels... {nf} found, {nm} missing, {ne} empty, {nc} corrupt"
tqdm(None, desc=prefix + d, total=n, initial=n) # display cache results
if cache['msgs']:
LOGGER.info('\n'.join(cache['msgs'])) # display warnings
Expand Down Expand Up @@ -523,7 +523,7 @@ def cache_labels(self, path=Path('./labels.cache'), prefix=''):
x[im_file] = [l, shape, segments]
if msg:
msgs.append(msg)
pbar.desc = f"{desc}{nf} found, {nm} missing, {ne} empty, {nc} corrupted"
pbar.desc = f"{desc}{nf} found, {nm} missing, {ne} empty, {nc} corrupt"

pbar.close()
if msgs:
Expand Down

0 comments on commit fd55271

Please sign in to comment.