Jetson inference with scikit learn

Autonomous Machines Jetson & Embedded Systems Jetson Nano
,
1

I currently using Jetson-inference. But I want to show the confusion matrix. How to run both jetson inference with scikit learn together?
Kindly provide a coding
Thank you for your kindness.

2

Hi @m012110024, the runtime programs from jetson-inference (like imagenet/imagenet.py, detectnet/detectnet.py, ect) only do inferencing and don’t support loading datasets for groundtruth. If desired, you could integrate scikit-learn with the PyTorch training scripts for calculation of these metrics during the training phase.

I haven’t used this, but it looks like there is an add-on library for PyTorch that computes the confusion matrix: https://torchmetrics.readthedocs.io/en/stable/classification/confusion_matrix.html

3

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