工程地址: https://github.com/yizt/Grad-CAM.pytorch ; 欢迎试用、关注并反馈问题...
pytorch 实现 Grad-CAM 和 Grad-CAM++,可以可视化任意分类网络的 Class Activation Map (CAM)图,包括自定义的网络;
同时测试了 VGG16,VGG19,ResNet50,ResNet101,DenseNet121,Inception_v3,MobileNetV2,ShuffleNetV2 网络在单个实例对象和多个实例对象上的表现;
在目标检测网络 faster r-cnn 上测试了 Grad-CAM 和 Grad-CAM++的表现,这是目前唯一一个在目标检测上实现了 Grad-CAM 和 Grad-CAM++的工程。