xchuwenbo / fuxi Goto Github PK

This is the official repository for the FuXi paper.

FuXi: A cascade machine learning forecasting system for 15-day global weather forecast

by Lei Chen, Xiaohui Zhong, Feng Zhang, Yuan Cheng, Yinghui Xu, Yuan Qi, Hao Li

The Google Drive folder contains the FuXi model, code, and sample input data, all of which are essential resources for this study. Access to these resources is restricted by a password-protected link. To obtain the password, users must complete the provided Google Form. For inquiries regarding the password, please contact Professor Li Hao at the email address: [email protected].

The downloaded files shall be organized as the following hierarchy:

├── root
│   ├── data
│   │    ├── 20210101
│   │         ├── input.nc
│   │         ├── output
│   │               ├── 006.nc
│   │               ├── 012.nc
│   │               ├── ...
│   │               ├── 360.nc
│   ├── model
│   |    ├── short
│   |    ├── short.onnx
│   |    ├── medium
│   |    ├── medium.onnx
│   |    ├── long
│   |    ├── long.onnx
|   |   
│   ├── fuxi.py
│   ├── util.py
│   ├── make_era5_input.py
│   ├── make_hres_input.py
│   ├── make_gfs_input.py

1. Install xarray

conda install -c conda-forge xarray dask netCDF4 bottleneck

2. Install onnxruntime

pip install -r requirement.txt

python fuxi.py --model model_dir --input input_file --num_steps 20 20 20

The input.nc file contains preprocessed data from the origin ERA5 files. The file has a shape of (2, 70, 721, 1440), where the first dimension represents two time steps. The second dimension represents all variable and level combinations, named in the following exact order:

'Z50', 'Z100', 'Z150', 'Z200', 'Z250', 'Z300', 'Z400', 'Z500', 'Z600', 'Z700', 'Z850', 'Z925', 'Z1000', 
'T50', 'T100', 'T150', 'T200', 'T250', 'T300', 'T400', 'T500', 'T600', 'T700', 'T850', 'T925', 'T1000', 
'U50', 'U100', 'U150', 'U200', 'U250', 'U300', 'U400', 'U500', 'U600', 'U700', 'U850', 'U925', 'U1000', 
'V50', 'V100', 'V150', 'V200', 'V250', 'V300', 'V400', 'V500', 'V600', 'V700', 'V850', 'V925', 'V1000', 
'R50', 'R100', 'R150', 'R200', 'R250', 'R300', 'R400', 'R500', 'R600', 'R700', 'R850', 'R925', 'R1000', 
'T2M', 'U10', 'V10', 'MSL', 'TP'

The last five variables ('T2M', 'U10', 'V10', 'MSL', 'TP') are surface variables, while the remaining variables represent atmosphere variables with numbers representing pressure levels.

NOTE:

• The variable 'Z' represents geopotential and not geopotential height.
• The variable 'TP' represents total precipitation accumulated over a period of 6 hours.