PyTorch implementation of a Neural Cellular Automaton as proposed by the paper "Growing Neural Cellular Automata", trained on medical images from MedMNIST and visualized using Pygame.
This project was done as part of the lecture Deep Generative Models at TU Darmstadt.
See requirements.txt
.
Run main.py
to start the interactive demonstration (no GPU required).
Edit and run schedule_training.py
or start_training.py
to train a new model (GPU recommended).
- Lukas Maninger
- Fabian Metschies
- Patrick Siebke
- Patrick Vimr