How to link the scikit-learn confusion matrix with jetson-inference image classification

So far I use jetson-inference but it cannot show the confusion matrix. How to link scikit-learn confusion matrix with jetson-inference? Can someone please show me the coding? Thank you

Hi @m012110024, as discussed in this thread, jetson-inference doesn’t load ground-truth data for datasets and hence it doesn’t compute the accuracy/ect. You could add this yourself, or instead use the original PyTorch models since PyTorch already has the dataloaders, and integrate that with sci-kit learn. Sorry I am unable to spend time doing the actual code for this right now.

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