Inferring image classification model in Jetson nano

Autonomous Machines Jetson & Embedded Systems Jetson TX2
,
1

I have trained image classification model on TensorFlow and received .h5 file, is there a way i can infer this model on jetson nano

3

Hi,

You can either run it with TensorFlow or convert the model into a TensorRT engine.

1. TensorFlow:
You can find the installation guide below:

2. TensorRT:
Please convert the model into ONNX format and then you can run the model with trtexec directly.

$ /usr/src/tensorrt/bin/trtexec --onnx=[your/model]

Below is an example for your reference:

Thanks.

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.