lambert-x / medical_mae

The official implementation of "Delving into Masked Autoencoders for Multi-Label Thorax Disease Classification"
Apache License 2.0
67 stars 15 forks source link

GradCAM #3

Closed BehzadBozorgtabar closed 1 year ago

BehzadBozorgtabar commented 1 year ago

Hi,

Can you please share your script for the GradCAM visualization for the ViT-s/16, figure 4, in the paper?

Thanks

lambert-x commented 1 year ago

Please refer to this notebook (the codes are a little bit dirty. Sorry about that) https://drive.google.com/file/d/1WG6zYHwaRbzTB0xbe28hKPJQmb0za8oT/view?usp=sharing