This is a code repository for training deep learning neural networks in physics using the SOFA environment. Deep Learning is applied to a liver model as an example. The repository is divided into four parts:
- Definition of the SOFA environment liver model with appropriate material properties
- Python code for generating data (batches) based on node displacement in SOFA simulations.
- A neural network design for learning and predicting strain.
- Visualization of model strain driven by the neural network.
Minimal workloop:
MAIN_Generate_scenarios.py --> PyTorch_learning.py --> network_PyTorch.pkl --> Liver_on_pygames.py