Running a custom image classification CNN

Autonomous Machines Jetson & Embedded Systems Jetson AGX Orin
1

Hi All,

I do have one custom image classification CNN model in python , How do I run this in target ? Is there is any examples or tutorial to follow, I am not sure how deep streamer library convert python to C or C++ and other details.

Will be helpful if I get a example to run a custom model prepared in python run on target

3

Hi,

Which framework do you use? PyTorch or TensorFlow?

Both frameworks can be installed on the Jetson. Please find the information below:
You can deploy it on the target in the same way as the desktop after installing the package.

https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html
https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html
Thanks.

5

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