- python 3.6.x or higher
- pytorch v1.0.0 (It might run well on v0.4.x, but I did not test yet.)
- pillow
- numpy
- matplotlib (pyplot)
python train.py
python test.py
python gradcam.py
- Test accuracy: 80.38%
- # of parameters: 2,940,682
| Correct Examples | |||
|---|---|---|---|
| Original Image |
 |
 |
 |
| Saliency Map |
 |
 |
 |
| Incorrect Examples | |||
|---|---|---|---|
| Original Image |
 |
 |
 |
| Saliency Map |
 |
 |
 |