We provide an easy way for visualizing
You will get a heatmap image when you send it the layer name, the model and the image.
the example code is located in the root, its name is 'Evison_Tutorial.ipynb'
We provide an example on goolge colab: https://colab.research.google.com/drive/1nNxHOur741WMjBIp1BHrmGi9xHb_CfDB?usp=sharing
You may specify which output you want to use for classification bythe argument target_output.
display = Display(network, visualized_layer, target_output=2, img_size=(224, 224)) # You may use the output of index 3 for classification.
https://github.com/utkuozbulak/pytorch-cnn-visualizations
Jones(Jinhong) Lin: jonneslin@gmail.com / jlin398@wisc.edu