From ebf3fc73cbdfb178ba07fd74b07dac410238fca8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 Aug 2021 15:48:53 +0200 Subject: [PATCH] Update 'results saved to' string (#4275) --- detect.py | 2 +- export.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/detect.py b/detect.py index 80517f342a41..88d1d9ca3800 100644 --- a/detect.py +++ b/detect.py @@ -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) diff --git a/export.py b/export.py index c98e92d972c6..83e293b72e73 100644 --- a/export.py +++ b/export.py @@ -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():