zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update 'results saved to' string (#4275)

This commit is contained in:
Glenn Jocher 2021-08-02 15:48:53 +02:00 committed by GitHub
parent 587c4b4b81
commit 388016e9e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -189,7 +189,7 @@ def run(weights='yolov5s.pt', # model.pt path(s)
if save_txt or save_img:
s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
print(f"Results saved to {save_dir}{s}")
print(f"Results saved to {colorstr('bold', save_dir)}{s}")
if update:
strip_optimizer(weights) # update model (to fix SourceChangeWarning)

4
export.py
View File

@ -155,7 +155,9 @@ def run(weights='./yolov5s.pt', # weights path
export_coreml(model, img, file)
# Finish
print(f'\nExport complete ({time.time() - t:.2f}s). Visualize with https://github.com/lutzroeder/netron.')
print(f'\nExport complete ({time.time() - t:.2f}s)'
f"Results saved to {colorstr('bold', file.parent.resolve())}\n"
f'Visualize with https://netron.app')
def parse_opt():