heatmap visualization of the whole picture

[zhaoxuyan]

Thanks for your brilliant work!
How can I get a heatmap visualization of the whole picture? Please don't just visualize the box part?

[gabrispada]

Maybe just change this line

yolov5-gradcam/main.py

Line 31 in 8ae1f58

 n_heatmat = (Box.fill_outer_box(heatmap, bbox) / 255).astype(np.float32) 

in this way:
n_heatmat = (heatmap / 255).astype(np.float32)