Skip to content
/ MedNCA Public

Neural Cellular Automata for Medical Images

Notifications You must be signed in to change notification settings

Pynator/MedNCA

Repository files navigation

MedNCA

license badge repo size badge files badge total lines badge

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.

GIF 1 GIF 2 GIF 3 GIF 4

Dependencies

See requirements.txt.

Usage

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).

Contributors

  • Lukas Maninger
  • Fabian Metschies
  • Patrick Siebke
  • Patrick Vimr

About

Neural Cellular Automata for Medical Images

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published