Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)
1.99k
stars
205
forks
source link
feat: Removes warnings for torchvision and matplotlib #224
Closed
frgfm closed 11 months ago
This PR removes the warnings on torchvision pretrained models and matplotlib colormaps.