Skip to content

ShihaoZhaoZSH/What-do-deep-nets-learn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commits
 
 
 
 
 
 

Repository files navigation

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages