├── README.md ├── data_gen.py ├── models ├── __init__.py ├── baselines │ ├── ConvLSTM.py │ ├── DHPM.py │ ├── GAN.py │ ├── ResNet.py │ ├── SST.py │ └── U-net.py ├── model.py └── penalty.py ├── notebooks ├── Evaluation.ipynb └── radialProfile.py ├── requirements.txt ├── run_model.py ├── utils ├── __init__.py ├── eval_utils.py └── train_utils.py └── videos ├── Ablation_TF.mp4 ├── U.mp4 ├── V.mp4 ├── all.gif └── model.png /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/README.md -------------------------------------------------------------------------------- /data_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/data_gen.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/baselines/ ConvLSTM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/baselines/ ConvLSTM.py -------------------------------------------------------------------------------- /models/baselines/DHPM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/baselines/DHPM.py -------------------------------------------------------------------------------- /models/baselines/GAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/baselines/GAN.py -------------------------------------------------------------------------------- /models/baselines/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/baselines/ResNet.py -------------------------------------------------------------------------------- /models/baselines/SST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/baselines/SST.py -------------------------------------------------------------------------------- /models/baselines/U-net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/baselines/U-net.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/model.py -------------------------------------------------------------------------------- /models/penalty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/models/penalty.py -------------------------------------------------------------------------------- /notebooks/Evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/notebooks/Evaluation.ipynb -------------------------------------------------------------------------------- /notebooks/radialProfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/notebooks/radialProfile.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/run_model.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /videos/Ablation_TF.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/videos/Ablation_TF.mp4 -------------------------------------------------------------------------------- /videos/U.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/videos/U.mp4 -------------------------------------------------------------------------------- /videos/V.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/videos/V.mp4 -------------------------------------------------------------------------------- /videos/all.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/videos/all.gif -------------------------------------------------------------------------------- /videos/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rose-STL-Lab/Turbulent-Flow-Net/HEAD/videos/model.png --------------------------------------------------------------------------------