pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to visualize multi-classes

O2andsAnoxia opened this issue · comments

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.

@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.