Jetson inference with scikit learn

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.

Hi @m012110024, the runtime programs from jetson-inference (like imagenet/, detectnet/, 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:

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