ndey96 / deep-generator-network

Unofficial implementation of "Synthesizing the preferred inputs for neurons in neural networks via deep generator networks" by Nguyen et al.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

deep-generator-network

Unofficial implementation of Synthesizing the preferred inputs for neurons in neural networks via deep generator networks by Nguyen et al..

Original: Screen Shot 2020-09-18 at 1 42 16 AM

Ours: Screen Shot 2020-09-18 at 1 42 08 AM

Pretrained models are available for download at: https://www.dropbox.com/sh/gdkr6n1d83jx2kg/AACet2EbEWFmlpysHNOSYemHa?dl=

To train, and generate new training weights, or synthesize using either the provided pre-trained checkpoints, or your own ilsvrc2012 is required local to your machine. Any hardcoded paths in train_parallel_deep*.py or synthesize.py, to ilsvrc2012 training and validation image directories should be set.

Various parameters can be set - you'll have to experiment and read the related papers to get a feel for that.

Python3, pytorch, pyvision, and tensorboard are required... any other requirements will error out... you can install them.

The code has not been fully tested on cpu only.

Training: python train_parallel_deepgen.py python train_parallel_deepsim.py

Synthesis: bash synth_figs.sh

About

Unofficial implementation of "Synthesizing the preferred inputs for neurons in neural networks via deep generator networks" by Nguyen et al.

License:MIT License


Languages

Language:Jupyter Notebook 92.0%Language:Python 7.4%Language:Shell 0.6%