/What-do-deep-nets-learn

What Do Deep Nets Learn? Class-wise Patterns Revealed in the Input Space

Primary LanguagePython

What Do Deep Nets Learn? Class-wise Patterns Revealed in the Input Space

Introduction:

  • Environment: Python3.6.5, PyTorch1.5.0
  • Dataset: CIFAR-10, ImageNet-1k

Usage:

This is our paper link. You can firstly run select_canvas.py to choose a canvas. Then you can run generateMap.py to generate the mask and generatePos.py for clip. Finally you can extract the pattern by extractPattern.py. test.py is used for calculating the predictive power.

Notes

  • Models used in our experiments are trained without normalization (i.e. torchvision.transforms.Normalization). To achieve this, for CIFAR-10 we just train from scratch and for ImageNet-1k, we fine-tune on the normalized trained model.
  • We reconstruct CIFAR-10 for convenience by generateSet.py.