search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.59k stars 510 forks source link

Create confusion matrix of class predictions #1457

Closed HannahAlexander closed 1 year ago

HannahAlexander commented 1 year ago

💡 Your Question

Hi,

I noticed with yolo-v8 it is possible to generate a confusion matrix. How can i do this with yolo-nas?

Thank you!

Versions

No response

BloodAxe commented 1 year ago

Currently this is not implemented/supported out of the box.

You do implement it manually as a custom metric or callback and log computed confusion matrix as an image to a logger.