Update 'results saved to' string (ultralytics#4275)

fhkiel-mlaip · Aug 2, 2021 · ebf3fc7 · ebf3fc7
1 parent 98e61a6
commit ebf3fc7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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():