UNet part of diffusion model provided to generate 3D polycrystal structures.
Generation sample:
Project repo: https://github.com/evgmaslov/structure_generation
Train code: https://github.com/evgmaslov/structure_generation/blob/main/train.ipynb
Inference code: https://github.com/evgmaslov/structure_generation/blob/main/inference.ipynb
- Downloads last month
- 8