Console corrupted -> corrupt (#6338)
* Console corrupted -> corrupt Minor style changes. * Update export.py
This commit is contained in:
1 parent
3119b2f27c
commit
fd55271c04
2 files changed
+4
-5
No files matched your search
@@ -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.'
|
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,
|
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,
|
agnostic_nms=agnostic_nms or tfjs, topk_per_class=topk_per_class,
|
||||||
topk_all=topk_all,
|
topk_all=topk_all, conf_thres=conf_thres, iou_thres=iou_thres) # keras model
|
||||||
conf_thres=conf_thres, iou_thres=iou_thres) # keras model
|
|
||||||
if pb or tfjs: # pb prerequisite to tfjs
|
if pb or tfjs: # pb prerequisite to tfjs
|
||||||
f = export_pb(model, im, file)
|
f = export_pb(model, im, file)
|
||||||
if tflite or edgetpu:
|
if tflite or edgetpu:
|
||||||
|
|||||||
+3
-3
@@ -424,9 +424,9 @@ class LoadImagesAndLabels(Dataset):
|
|||||||
cache, exists = self.cache_labels(cache_path, prefix), False # cache
|
cache, exists = self.cache_labels(cache_path, prefix), False # cache
|
||||||
|
|
||||||
# Display 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:
|
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
|
tqdm(None, desc=prefix + d, total=n, initial=n) # display cache results
|
||||||
if cache['msgs']:
|
if cache['msgs']:
|
||||||
LOGGER.info('\n'.join(cache['msgs'])) # display warnings
|
LOGGER.info('\n'.join(cache['msgs'])) # display warnings
|
||||||
@@ -523,7 +523,7 @@ class LoadImagesAndLabels(Dataset):
|
|||||||
x[im_file] = [l, shape, segments]
|
x[im_file] = [l, shape, segments]
|
||||||
if msg:
|
if msg:
|
||||||
msgs.append(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()
|
pbar.close()
|
||||||
if msgs:
|
if msgs:
|
||||||
|
|||||||
Reference in new issue
Block a user