├── .gitignore ├── README.md ├── datasets ├── __init__.py ├── dataset_cikm.py ├── dataset_meteonet.py ├── dataset_sevir.py ├── dataset_shanghai.py ├── get_datasets.py └── preprocess.py ├── env.yaml ├── framework.png ├── models ├── alphapre.py └── simvp │ ├── __init__.py │ └── simvp_iter.py ├── run.py └── utils ├── metrics.py ├── metrics_valid.py └── tools.py /.gitignore: -------------------------------------------------------------------------------- 1 | Exps/ 2 | resources/ 3 | __pycache__/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/dataset_cikm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/datasets/dataset_cikm.py -------------------------------------------------------------------------------- /datasets/dataset_meteonet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/datasets/dataset_meteonet.py -------------------------------------------------------------------------------- /datasets/dataset_sevir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/datasets/dataset_sevir.py -------------------------------------------------------------------------------- /datasets/dataset_shanghai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/datasets/dataset_shanghai.py -------------------------------------------------------------------------------- /datasets/get_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/datasets/get_datasets.py -------------------------------------------------------------------------------- /datasets/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/datasets/preprocess.py -------------------------------------------------------------------------------- /env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/env.yaml -------------------------------------------------------------------------------- /framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/framework.png -------------------------------------------------------------------------------- /models/alphapre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/models/alphapre.py -------------------------------------------------------------------------------- /models/simvp/__init__.py: -------------------------------------------------------------------------------- 1 | from .simvp_iter import get_model -------------------------------------------------------------------------------- /models/simvp/simvp_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/models/simvp/simvp_iter.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/run.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/metrics_valid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/utils/metrics_valid.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linkenghong/AlphaPre/HEAD/utils/tools.py --------------------------------------------------------------------------------