glenn-jocher commited on
Commit
388016e
1 Parent(s): 587c4b4

Update 'results saved to' string (#4275)

Browse files
Files changed (2) hide show
  1. detect.py +1 -1
  2. export.py +3 -1
detect.py CHANGED
@@ -189,7 +189,7 @@ def run(weights='yolov5s.pt', # model.pt path(s)
189
 
190
  if save_txt or save_img:
191
  s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
192
- print(f"Results saved to {save_dir}{s}")
193
 
194
  if update:
195
  strip_optimizer(weights) # update model (to fix SourceChangeWarning)
 
189
 
190
  if save_txt or save_img:
191
  s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
192
+ print(f"Results saved to {colorstr('bold', save_dir)}{s}")
193
 
194
  if update:
195
  strip_optimizer(weights) # update model (to fix SourceChangeWarning)
export.py CHANGED
@@ -155,7 +155,9 @@ def run(weights='./yolov5s.pt', # weights path
155
  export_coreml(model, img, file)
156
 
157
  # Finish
158
- print(f'\nExport complete ({time.time() - t:.2f}s). Visualize with https://github.com/lutzroeder/netron.')
 
 
159
 
160
 
161
  def parse_opt():
 
155
  export_coreml(model, img, file)
156
 
157
  # Finish
158
+ print(f'\nExport complete ({time.time() - t:.2f}s)'
159
+ f"Results saved to {colorstr('bold', file.parent.resolve())}\n"
160
+ f'Visualize with https://netron.app')
161
 
162
 
163
  def parse_opt():