How to visualize multi-classes

pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam

MIT License

281 stars 47 forks source link

How to visualize multi-classes #1

Closed O2andsAnoxia closed 2 years ago

O2andsAnoxia commented 2 years ago

Hi, Thanks for your great work and the results look amazing. But I wonder that if it can visualize multi-classes by grad-cam whatever in classification or detection task.

pooya-mohammadi commented 2 years ago

@O2andsAnoxia Hi, it's my pleasure. As far as I'm aware, GradCam and other gradient-based visualization algorithms maximize the output of a single class, not altogether. If you have any different algorithms in mind, inform me, please.