Minimal inference example

Robotics & Edge Computing Jetson Systems Jetson Nano
1

Hi,

I’ve been working on the getting started with jetson nano dli course. I’ve developed a great custom pythorch resnet-34 model that i would like to run in a python file by itself. So super simple: i have a file .pth and i would like to:

  1. Load model
  2. Grab image
  3. Print class + confidence

Are there any examples showing this simple setup?

Thanks in advance!

3

Hi,

On Nano, it’s recommended to use TensorRT for inference.
Please convert your pth model into the ONNX format first.

After that, you can check the below sample to read images (using OpenCV) and print the model output.
https://elinux.org/Jetson/L4T/TRT_Customized_Example#OpenCV_with_ONNX_model

Thanks.