jjxxmiin / DeepVisual_QTorch

Deep Learning Visualization Tools Using PyTorch

Visual_QTorch

Deep Learning Visualization Tools Using PyTorch

Requirement

• Pytorch 1.1.0
• PyQt5

pip install PyQt5

Demo

• CAM

• Grad CAM

How to use

python view.py

1. Upload Image
2. Select Model
3. Select Mode
4. Select Class Number(model predict : -1)
5. Click Start

Support Dataset

• ImageNet

Support Model

• Resnet18

Support Vis

• CAM
• Grad CAM
• Guided Backprop
• Vanilla Grad
• Smooth Grad (Slow..)

TODO

• ADD Support Model
• ADD Support Visualization
• ADD Support Dataset
• ...

Reference:

• https://gist.github.com/yrevar/942d3a0ac09ec9e5eb3a
• https://github.com/utkuozbulak/pytorch-cnn-visualizations