KO
|
EN
gitlite — search
Search
#python
#java
#python3
#arduino
#golang
#machine-learning
#rust
#html
#flask
#javascript
#seismology
#nodejs
STFormer
★ 24
Open GitHub ↗
基于Transformer的雷达回波外推用于近期降水预测的研究
Download README (.md)
Explore Similar Repositories
Fundamentals-of-Analog-Electronic-Techniques
:
模拟电子技术基础
lastfm-scrobbler
:
🎵 A Last.fm scrobbler via MPRIS2 in Linux, designed for media players without built-in support for Last.fm
f3rm.github.io
:
F3RM: Feature Fields for Robotic Manipulation - Website
fastlane-plugin-rustore
:
Fastlane плагин для заливки приложений в Rustore
AD-Ray-Optics
:
Auto Differentiable Ray Tracing Framework
// repository documentation
Was this content helpful?
★ 0
(0 ratings)
Select Rating:
★
★
★
★
★
Submit Feedback
Recent Feedback
×
Download README
Do you want to download the
README.md
file for
STFormer
?
Download (.md)
# 基于Transformer的雷达回波外推用于近期降水预测的研究 ## 代码模块 `config`:包含配置文件 `script`:包含运行脚本 `simvp/api`:包含一个实验的启动器 `simvp/core`:包含核心的运行插件和基准测试 `simvp/dataset`:包含数据集和数据加载器 `simvp/method`:包含simvp方法的实现 `simvp/model`:包含STFormer模型的实现 `simvp/modules`:包含核心组件的实现 `simvp/utils`:包含工具函数 ## 数据集 适用于任何时空数据 例如: 2000条雷达回波数据集, 格式`T x C x H x W`,其中`T`为时间维度,`H`为高度维度,`W`为宽度维度, `C`为通道维度,`C=1`,表示仅雷达。 ## 配置环境 1. 创建anaconda环境 `conda create -n simvp python=3.8` 2. 安装pytorch `conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia` 需要与系统cuda版本对应 3. 安装其他依赖 `pip install -r requirements.txt` ## 运行 1. 修改train_radar_gsta.sh,设置数据集路径、模型保存路径等 2. 运行 `bash script/train_radar_gsta.sh` ## 推理 1. 修改infer_radar_gsta.sh,设置权重路径、数据集路径等 2. 运行 `bash script/infer_radar_gsta.sh`