Parametric generation of conditional geological realizations using generative neural networks (arXiv)
Requires PyTorch 0.4+
Run with python main.py [--options]
- Z
- X
- O
You can download our pre-trained unconditional generator netG.pth
here (12MB), which has been trained using Wasserstein GAN (https://arxiv.org/abs/1701.07875)
main.py
main codemodels.py
neural network architecturesutils.py
helper functionsdat/cond*.dat
conditioning test cases