How to draw the grad-cam like paper Fig.5?

eeyhsong / NICE-EEG

[ICLR 2024] M/EEG-based image decoding with contrastive learning. i. Propose a contrastive learning framework to align image and eeg. ii. Resolving brain activity for biological plausibility.

https://arxiv.org/abs/2308.13234

MIT License

73 stars 10 forks source link

How to draw the grad-cam like paper Fig.5? #6

Open ChiShengChen opened 2 months ago

ChiShengChen commented 2 months ago

2024-05-03 16-33-03 的螢幕擷圖

I have got the Enc_eeg_cls.pth and Proj_eeg_cls.pth weights, could you tell me how to drow the Grad-Cam?

Thanks for big help

EvelynChengG commented 2 months ago

Do you know what is "frontral","central" in "spatial.py"? These variables do not appear to be assigned values