evgmaslov/diffusion-3d-material

UNet part of diffusion model provided to generate 3D polycrystal structures.\ Generation sample:

Project repo: https://github.com/evgmaslov/structuregeneration \ Train code: https://github.com/evgmaslov/structuregeneration/blob/main/train.ipynb \ Inference code: https://github.com/evgmaslov/structure_generation/blob/main/inference.ipynb