├── .gitignore ├── README.md ├── configs ├── __init__.py ├── dataset.yaml ├── end2end.yaml ├── transmodel.yaml └── warmup.yaml ├── data_generation ├── README.md ├── create_physics_scenes.py ├── create_surface_meshes.py ├── models │ ├── Box.obj │ └── Fluid_default.obj ├── physics_data_helper.py └── splishsplash_config.py ├── datasets ├── dataset.py └── dataset_splishsplash_rawdata.py ├── eval_e2e.py ├── eval_renderer.py ├── eval_transmodel.py ├── models ├── nerf.py ├── renderer.py └── transmodel.py ├── requirements.txt ├── run.sh ├── train_e2e.py ├── train_renderer.py ├── train_transmodel.py ├── trainer ├── __init__.py ├── basetrainer.py ├── trainer_e2e.py ├── trainer_renderer.py └── trainer_transmodel.py └── utils ├── evaluate_images.ipynb ├── lr_schedulers.py ├── particles_utils.py ├── point_eval.py └── ray_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/configs/dataset.yaml -------------------------------------------------------------------------------- /configs/end2end.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/configs/end2end.yaml -------------------------------------------------------------------------------- /configs/transmodel.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/configs/transmodel.yaml -------------------------------------------------------------------------------- /configs/warmup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/configs/warmup.yaml -------------------------------------------------------------------------------- /data_generation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/README.md -------------------------------------------------------------------------------- /data_generation/create_physics_scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/create_physics_scenes.py -------------------------------------------------------------------------------- /data_generation/create_surface_meshes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/create_surface_meshes.py -------------------------------------------------------------------------------- /data_generation/models/Box.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/models/Box.obj -------------------------------------------------------------------------------- /data_generation/models/Fluid_default.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/models/Fluid_default.obj -------------------------------------------------------------------------------- /data_generation/physics_data_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/physics_data_helper.py -------------------------------------------------------------------------------- /data_generation/splishsplash_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/data_generation/splishsplash_config.py -------------------------------------------------------------------------------- /datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/datasets/dataset.py -------------------------------------------------------------------------------- /datasets/dataset_splishsplash_rawdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/datasets/dataset_splishsplash_rawdata.py -------------------------------------------------------------------------------- /eval_e2e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/eval_e2e.py -------------------------------------------------------------------------------- /eval_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/eval_renderer.py -------------------------------------------------------------------------------- /eval_transmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/eval_transmodel.py -------------------------------------------------------------------------------- /models/nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/models/nerf.py -------------------------------------------------------------------------------- /models/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/models/renderer.py -------------------------------------------------------------------------------- /models/transmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/models/transmodel.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/run.sh -------------------------------------------------------------------------------- /train_e2e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/train_e2e.py -------------------------------------------------------------------------------- /train_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/train_renderer.py -------------------------------------------------------------------------------- /train_transmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/train_transmodel.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainer/basetrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/trainer/basetrainer.py -------------------------------------------------------------------------------- /trainer/trainer_e2e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/trainer/trainer_e2e.py -------------------------------------------------------------------------------- /trainer/trainer_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/trainer/trainer_renderer.py -------------------------------------------------------------------------------- /trainer/trainer_transmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/trainer/trainer_transmodel.py -------------------------------------------------------------------------------- /utils/evaluate_images.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/utils/evaluate_images.ipynb -------------------------------------------------------------------------------- /utils/lr_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/utils/lr_schedulers.py -------------------------------------------------------------------------------- /utils/particles_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/utils/particles_utils.py -------------------------------------------------------------------------------- /utils/point_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/utils/point_eval.py -------------------------------------------------------------------------------- /utils/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/syguan96/NeuroFluid/HEAD/utils/ray_utils.py --------------------------------------------------------------------------------