SimronThapa / FSRN-CVPR2020

This codebase implements the system described in the paper: Dynamic Fluid Surface Reconstruction Using Deep Neural Network
https://ivlab.cse.lsu.edu/FSRN_CVPR20.html
MIT License
24 stars 10 forks source link
3d-reconstruction convolutional-neural-networks deep-learning fluid-dynamics recurrent-neural-networks tensorflow

Dynamic Fluid Surface Reconstruction Using Deep Neural Network

Web Page | Paper

Simron Thapa, Nianyi Li, Jinwei Ye, Imaging and Vision Lab, Louisiana State University. In CVPR 2020 (oral).

We present a dynamic fluid surface reconstruction network that recovers time-varying 3D fluid surfaces from a single viewpoint.

Contributions

  1. We design physics-motivated loss functions for network training
  2. We synthesize a large fluid dataset using physics-based modeling and rendering [Check out the folder "Fluid_wave_simulator". It is our synthetic data generation MatLab code.]
  3. Our network is validated on real captured fluid data

Datasets

Complete Training data will be made available soon.

Training, Validation, Testing

Please remember to cite the paper if you use this dataset.

Training and Testing

The data preprocessing code (before training with FSRN-CNN) and data post-processing code for the predictions (before training with FSRN-RNN) will be made available soon.

python FSRN-CNN-train.py
python FSRN-RNN-train.py

Evaluation

The code for evaluating the predictions with ground truth values. We use accuracy and error matrics.

python evaluate_metrics.py

Results

  1. Synthetic Results

  1. Real Results

  1. Re-rendered Results

Citation

If you find this work useful, please consider citing:

@InProceedings{Thapa_2020_CVPR,
author = {Thapa, Simron and Li, Nianyi and Ye, Jinwei},
title = {Dynamic Fluid Surface Reconstruction Using Deep Neural Network},
booktitle = {The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}