3d-Gradcam

在中文互联网中并没有找到3d-cam可视化的资料，monai中的3d可视化不适合我的需求，于是自己修改了3d-Gradcam。初始官方代码是https://github.com/jacobgil/pytorch-grad-cam/tree/master

主要修改了：

• grad_cam.py中的get_cam_weights，将axis=(2, 3)修改为axis=(3, 4)，适应3d的维度
• base_cam.py中的BaseCAM基类中的get_cam_image方法，指定输出weights这个3d数组中的特定2d维度子数组。
• 由于自己特定层的输出并不是一个单一数组，还修改了ActivationsAndGradients中的output
• 可以根据自己的数据特点修改image.py中的show_cam_on_image
• test.py

test.py

是利用gradcam调用unet中特定层输出可视化图的文件 GradCAM也要指定输出层