#365 · Primary category: Computer Vision

gandissect

deep-learning gan generative-adversarial-network image-manipulation interactive-visualizations interpretable-ml pytorch

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/

Project last updated:05/23/21

GitHub Stars

1.8K

Forks

276

Contributors

3

License

MIT

Why we included this project

If you train or fine-tune image generation GANs and wonder which internal units actually fire for trees, windows, or doors, GANDissect answers that directly. It maps the neurons of a generative network onto human-interpretable concepts, then lets you inspect the results as a static summary or through the interactive GANPaint demo, where you can draw new objects into a scene and watch the model respond. The same unit-level control lets you diagnose artifacts and remove or insert features, like erasing windows from a room, by manipulating the units that produce them. This is research code tied to a 2018 paper, so expect a conda setup and a CUDA GPU rather than a polished application, but for anyone doing careful analysis of why a GAN renders the way it does, it is still the clearest option around.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category