FuXi-Ocean 提供基于 ONNX 的海洋预报推理和 RMSE 评估代码,支持 HYCOM 格式数据,被 NeurIPS 2025 接收为口头报告。

Stars

24

7 天增长

暂无数据

Fork 数

0

开放 Issue

0

开源协议

暂无数据

最近更新

2026-06-27

AI 仓库情报摘要
FR-AI / ANALYSIS

为什么值得关注

作为 NeurIPS 2025 口头报告的开源海洋预报模型,它提供了面向高分辨率全球海洋场的生产级 ONNX 推理管线。

适合谁使用

  • 海洋学家和气候科学家
  • 地球科学领域的机器学习研究人员
  • 业务化海洋预报人员
  • 从事地理空间时间序列数据分析的数据科学家

典型使用场景

  • 以 1/12° 分辨率进行全球海洋状态预报
  • 基于 HYCOM 再分析数据对 ML 海洋模型进行基准测试
  • 针对物理系统的自回归深度学习研究
  • 为大规模地球科学部署 ONNX 提供教学示例

项目优势

  • 被 NeurIPS 2025 接收为口头报告,认可度高
  • 提供完整的推理和评估脚本,文档清晰
  • 使用 ONNX Runtime,支持 CPU/GPU 高效推理并给出资源需求
  • 支持以 6 小时间隔进行 40 步的自回归预报

使用前须知

  • ONNX 权重文件尚未发布(计划中)
  • 全分辨率运行需 128 GB 以上内存和 32 核 CPU,限制可访问性
  • 假设使用特定 Zarr 格式的预归一化 HYCOM 数据,未提供训练代码

README 快速开始

FuXi-Ocean

Official open-source inference code for FuXi-Ocean.

FuXi-Ocean has been accepted as an Oral presentation at NeurIPS 2025. See the NeurIPS virtual page: https://neurips.cc/virtual/2025/loc/san-diego/poster/117603

This repository provides ONNX-based ocean forecasting inference and RMSE evaluation utilities for HYCOM-format ocean data.

Release Plan

  • Inference code
  • Test code
  • ONNX weight files

Repository Layout

.
├── inference_hycom_aim_full.py  # ONNX inference script
├── infer_hycom_full.sh          # Example inference launcher
├── eval_rmse.py                 # RMSE evaluation script
├── run_eval_rmse.sh             # Example evaluation launcher
├── re.txt                       # Python dependencies
└── README.md                    # Project documentation

Environment

Install the Python dependencies:

pip install -r re.txt

Recommended runtime requirements:

  • Python 3.8+
  • 32+ CPU cores for CPU inference
  • 128 GB+ memory for full-resolution HYCOM fields

Inference

inference_hycom_aim_full.py runs autoregressive ONNX inference on normalized HYCOM ocean data.

Arguments

ArgumentTypeDefaultDescription
--inputstrrequiredInput Zarr dataset path
--modelstrrequiredONNX model path
--maskstrrequiredLand-mask NetCDF path
--outputstrrequiredOutput directory
--devicestrcudaONNX Runtime execution provider preference: cuda or cpu
--total_stepint40Number of autoregressive forecast steps
--hour_intervalint6Forecast interval in hours
--start_timestr20150101Inclusive start time
--end_timestr20151230Inclusive end time
--strideint1Stride over candidate initialization windows

Input Data

The expected input is a normalized Zarr dataset with dimensions (time, channel, lat, lon).

Example structure:

hycom.6h.2006_2015.2160_4320.c105/
├── .zarray
├── .zattrs
├── channel/
├── lat/
├── lon/
├── time/
├── data/
├── mean.nc
├── std.nc
└── land_mask.nc

Expected data shape:


Dimensions:  (time: 14044, channel: 105, lat: 2160, lon: 4320)
Coordinates:
  * time     (time) datetime64[ns] 2006-01-01T12:00:00 ... 2015-12-31
  * channel  (channel) 
Coordinates:
  * channel  (chan

项目描述

Ocean forecasting and RMSE evaluation scripts.

相关仓库与替代方案

根据分类、Topic 和编程语言匹配的相似项目。

lopopolo
精选
lopopolo GitHub avatar

harness-engineering

Harness Engineering is a methodology for improving coding agent outputs by carefully crafting the environment around them—providing curated context, tools, and executable constraints that encode an organization’s nonfunctional requirements and cumulative lessons.

AI 与机器学习AI 智能体
2,390
slvDev
精选
slvDev GitHub avatar

esp32-ai

A 28.9 million parameter language model runs on an $8 ESP32-S3 microcontroller entirely on-device, generating simple stories at about 9.5 tokens per second.

AI 与机器学习大语言模型
1,960
littledivy
精选
littledivy GitHub avatar

mimic

mimic captures traffic from any iOS or web app and automatically generates a Python client library that lets you call the app's API like a regular library.

AI 与机器学习
1,482