glenn-jocher commited on
Commit
5e4ff19
1 Parent(s): b884ea3

Improved `export.py` usage examples (#6495)

Browse files

* Improved `export.py` usage examples

* Cleanup

Files changed (1) hide show
  1. export.py +4 -4
export.py CHANGED
@@ -469,10 +469,10 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path'
469
  f = [str(x) for x in f if x] # filter out '' and None
470
  LOGGER.info(f'\nExport complete ({time.time() - t:.2f}s)'
471
  f"\nResults saved to {colorstr('bold', file.parent.resolve())}"
472
- f"\nVisualize with https://netron.app"
473
- f"\nDetect with `python detect.py --weights {f[-1]}`"
474
- f" or `model = torch.hub.load('ultralytics/yolov5', 'custom', '{f[-1]}')"
475
- f"\nValidate with `python val.py --weights {f[-1]}`")
476
  return f # return list of exported files/dirs
477
 
478
 
 
469
  f = [str(x) for x in f if x] # filter out '' and None
470
  LOGGER.info(f'\nExport complete ({time.time() - t:.2f}s)'
471
  f"\nResults saved to {colorstr('bold', file.parent.resolve())}"
472
+ f"\nDetect: python detect.py --weights {f[-1]}"
473
+ f"\nPyTorch Hub: model = torch.hub.load('ultralytics/yolov5', 'custom', '{f[-1]}')"
474
+ f"\nValidate: python val.py --weights {f[-1]}"
475
+ f"\nVisualize: https://netron.app")
476
  return f # return list of exported files/dirs
477
 
478