项目作者: Nikhil-Kasukurthi

项目描述 :
Using Grad-cam activations to visualise CNN models on Pytorch
高级语言: Python
项目地址: git://github.com/Nikhil-Kasukurthi/CNN-Model-Visualisation.git


CNN-Model-Visualisation

Using Grad-cam activations to visualise CNN models on Pytorch

This is the supporting repository for the Visualising CNN Models Using Pytorch article.

Required dependencies

OpenCV
pip install opencv_python

PyTorch

If you are going to use a pretrained model from PyTorch, install torchvision as well

Results on the ISIC Skin cancer challenge.

Input Output
Input Grad Cam