| # FuXi-CFD |
|
|
| This repository contains: |
|
|
| 1) **Model weights** (ONNX) under `model/` |
| 2) A **runnable inference example** under `inference_example/` |
|
|
| ## Quick start (inference example) |
|
|
| ```bash |
| cd inference_example |
| python scripts/infer.py \ |
| --model ../model/fuxicfd_model.onnx \ |
| --input data/inputs.npz \ |
| --output data/prediction.npz |
| ``` |
|
|
| Outputs are saved as `data/prediction.npz` with keys: `u, v, w, k` (each shape `(27, 300, 300)`). |
|
|
| ## Folder structure |
|
|
| - `model/`: exported ONNX weights + model documentation |
| - `inference_example/`: end-to-end preprocessing → ONNX → postprocessing demo |
| - `data/`: sample input/output |
| - `normalization/`: scaler parameters and documentation |
| - `scripts/`: runnable scripts and notebooks |
| - `utils/`: reusable helper functions |
|
|
| ## License |
|
|
| This project is licensed under the Creative Commons |
| Attribution-NonCommercial 4.0 International License (CC BY-NC 4.0). |
| See the LICENSE file for details. |
|
|
