Rayleigh_Benard_Convection
收藏Rayleigh-Bénard Flow Dataset 概述
数据集基本信息
- 许可证: MIT
- 语言: 英语
- 标签: 物理学、偏微分方程、多物理场
数据集内容
- 包含高分辨率直接数值模拟数据,模拟不同瑞利数下的Rayleigh-Bénard对流
- 数据包括128×128网格上的速度场和温度分布
- 文件格式: .npz
数据字段
vx: 速度场的x分量vy: 速度场的y分量temp: 温度场time: 时间点
可用案例
-
Ra = 12,000
- 网格大小: 128×128
- VX范围: -18.24到23.19
- VY范围: -31.27到19.46
- 温度范围: -1.63到1.01
- 帧数: 500
-
Ra = 20,000
- 网格大小: 128×128
- VX范围: -24.74到32.24
- VY范围: -40.23到27.77
- 温度范围: -1.83到1.52
- 帧数: 501
安装与下载
bash pip install -r requirements.txt
python from huggingface_hub import hf_hub_download import os os.makedirs("./rb_data", exist_ok=True)
hf_hub_download( repo_id="your-username/rb-flow-visualization", filename="data/data_12e3.npz", local_dir="./rb_data", repo_type="dataset" )
数据集加载器
- RBFlowDataset: PyTorch Dataset类,处理.npz文件的加载和预处理
- load_rb_flow_data: 创建DataLoader的实用函数
数据统计访问
python dataset = RBFlowDataset(data/data_12e3.npz) stats = dataset.stats
基本用法
python from dataloader import load_rb_flow_data dataloader, dataset = load_rb_flow_data( data_path=data/data_12e3.npz, batch_size=32, shuffle=True )
加载序列
python from dataloader import RBFlowDataset dataset = RBFlowDataset(data/data_12e3.npz) sequence = dataset.get_sequence(start_idx=0, length=10)
测试
- 包含全面的测试套件(
test_functionality.py) - 测试内容包括数据集加载、数据处理和可视化
可视化工具
- 包括创建流场各种可视化的工具
创建动画
python from visualize import RBFlowVisualizer viz = RBFlowVisualizer(data/data_12e3.npz) viz.create_velocity_animation(output_path=velocity_animation.gif, fps=30, skip=3)
引用
bibtex @article{rahman2024pretraining, title={Pretraining Codomain Attention Neural Operators for Solving Multiphysics PDEs}, author={Rahman, Md Ashiqur and George, Robert Joseph and Elleithy, Mogab and Leibovici, Daniel and Li, Zongyi and Bonev, Boris and White, Colin and Berner, Julius and Yeh, Raymond A and Kossaifi, Jean and Azizzadenesheli, Kamyar and Anandkumar, Anima}, journal={Advances in Neural Information Processing Systems}, volume={37} year={2024} }




