Skip to content

Instantly share code, notes, and snippets.

@Aldhanekaa

Aldhanekaa/Jetson Nano Train Model cheatsheet

Created March 16, 2022 04:32
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save Aldhanekaa/987c9789ed314170249e40461a575c6f to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save Aldhanekaa/987c9789ed314170249e40461a575c6f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Jetson Nano Train Model cheatsheet
568 python3 train_ssd.py --dataset-type=voc --data=data/ftc-objects --model-dir=models/ftc-objects --batch-size=1 --workers=1 --epochs=1
569 python3 onnx_export.py --model-dir=models/ftc-objects/
570 detectnet --model=models/ftc-objects/ssd-mobilenet.onnx --labels=models/ftc-object/labels.txt --input-blob= input_0 --output-bbox=boxes /dev/video0
@Aldhanekaa
Copy link
Author

Aldhanekaa commented Mar 17, 2022

python3 train_ssd.py --dataset-type=voc --data=data/FTC-Objects-Model-Library --model-dir=models/FTC-Objects-Model-Library --batch-size=1 --workers=1 --epochs=1

@Aldhanekaa
Copy link
Author

Aldhanekaa commented Mar 17, 2022

python3 onnx_export.py --model-dir=models/FTC-Objects-Model-Library

@Aldhanekaa
Copy link
Author

Aldhanekaa commented Jun 2, 2022

detectnet --model=models/CT2/ssd-mobilenet.onnx --labels=models/CT2/labels.txt --input-blob=input_0 --output-cvg=scores --threshold=0.4 --output-bbox=boxes /dev/video0

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment