ShuaiLYU / grad_cam_pytorch

PyTorch implementation of Grad-CAM with hook

grad_cam_pytorch

PyTorch implementation of Grad-CAM with hook

Note:

• Fork from https://github.com/jacobgil/pytorch-grad-cam
• Batch input enabled.
• Multi-layer prediction enabled.
• implementation with hook.