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