Skip to content

Instantly share code, notes, and snippets.

@jogam5

jogam5/subprocess_tao_nvidia.py

Created December 19, 2022 05:39
Show Gist options
  • Save jogam5/11cc37b2d87b1af11428761ea4ccbfaf to your computer and use it in GitHub Desktop.
Save jogam5/11cc37b2d87b1af11428761ea4ccbfaf to your computer and use it in GitHub Desktop.
Download ZIP
subprocesses python library
Raw
subprocess_tao_nvidia.py
THIS WORKS!
import subprocess
cmd = ['tao', 'yolo_v4_tiny', 'export', '-m',
'/workspace/tao-experiments/yolo_v4_tiny/experiment_dir_retrain/weights/yolov4_cspdarknet_tiny_epoch_005.tlt',
'-k', 'YOUR_KEY',
'-e', '/workspace/tao-experiments/yolo_v4_tiny/specs/yolo_v4_tiny_retrain_kitti.txt',
'-o', '/workspace/tao-experiments/yolo_v4_tiny/export/yolov4_cspdarknet_tiny_epoch_005.etlt',
'--data_type', 'fp32',
'--gen_ds_config']
output = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)#.stdout
print(output)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment