├── DownScale_Paper ├── DownScale_Correction_split │ ├── data.txt │ ├── train_12_36.txt │ └── val_12_36.txt └── param_stat_12_36 │ ├── DPT_2_m_stat.pkl │ ├── HGT_obs_stat.pkl │ ├── HGT_stat.pkl │ ├── SP_stat.pkl │ ├── T_2_m_obs_stat.pkl │ ├── T_2_m_stat.pkl │ ├── U_10_m_stat.pkl │ ├── V_10_m_stat.pkl │ ├── WS_10_m_obs_stat.pkl │ └── WS_10_m_stat.pkl ├── LICENSE ├── Plot └── pcolor_map_one.py ├── README.md ├── VERSION ├── archs ├── __init__.py ├── climate_rcan_arch.py ├── climate_swinir_arch.py ├── climate_uformer_arch.py ├── climate_unet_arch.py ├── uformer_module.py └── utils.py ├── data ├── __init__.py ├── merge_dataset.py └── transforms.py ├── metrics ├── __init__.py ├── calculate_climate_mae_mse.py └── calculate_climate_psnr_ssim.py ├── models ├── __init__.py ├── climatesr_model.py ├── climateswinir_model.py └── climateuformer_model.py ├── paper_options ├── RCAN_multivarall_zscore_option.yml ├── SwinIR_c180_6x6_multivarall_zscore_option.yml ├── UNet_multivarall_zscore_option.yml ├── Uformer_MultiFuseOut_multivarall_zscore_option.yml ├── Uformer_MultiScaleHGT_MultiFuseOut_multivarall_zscore_option.yml ├── Uformer_MultiScaleHGT_multivarall_zscore_option.yml └── Uformer_multivarall_zscore_option.yml ├── requirements.txt ├── run.sh ├── setup.cfg ├── test.py └── train.py /DownScale_Paper/DownScale_Correction_split/data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/DownScale_Correction_split/data.txt -------------------------------------------------------------------------------- /DownScale_Paper/DownScale_Correction_split/train_12_36.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/DownScale_Correction_split/train_12_36.txt -------------------------------------------------------------------------------- /DownScale_Paper/DownScale_Correction_split/val_12_36.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/DownScale_Correction_split/val_12_36.txt -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/DPT_2_m_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/DPT_2_m_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/HGT_obs_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/HGT_obs_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/HGT_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/HGT_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/SP_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/SP_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/T_2_m_obs_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/T_2_m_obs_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/T_2_m_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/T_2_m_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/U_10_m_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/U_10_m_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/V_10_m_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/V_10_m_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/WS_10_m_obs_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/WS_10_m_obs_stat.pkl -------------------------------------------------------------------------------- /DownScale_Paper/param_stat_12_36/WS_10_m_stat.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/DownScale_Paper/param_stat_12_36/WS_10_m_stat.pkl -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/LICENSE -------------------------------------------------------------------------------- /Plot/pcolor_map_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/Plot/pcolor_map_one.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/README.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/__init__.py -------------------------------------------------------------------------------- /archs/climate_rcan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/climate_rcan_arch.py -------------------------------------------------------------------------------- /archs/climate_swinir_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/climate_swinir_arch.py -------------------------------------------------------------------------------- /archs/climate_uformer_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/climate_uformer_arch.py -------------------------------------------------------------------------------- /archs/climate_unet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/climate_unet_arch.py -------------------------------------------------------------------------------- /archs/uformer_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/uformer_module.py -------------------------------------------------------------------------------- /archs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/archs/utils.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/merge_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/data/merge_dataset.py -------------------------------------------------------------------------------- /data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/data/transforms.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/calculate_climate_mae_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/metrics/calculate_climate_mae_mse.py -------------------------------------------------------------------------------- /metrics/calculate_climate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/metrics/calculate_climate_psnr_ssim.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/climatesr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/models/climatesr_model.py -------------------------------------------------------------------------------- /models/climateswinir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/models/climateswinir_model.py -------------------------------------------------------------------------------- /models/climateuformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/models/climateuformer_model.py -------------------------------------------------------------------------------- /paper_options/RCAN_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/RCAN_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /paper_options/SwinIR_c180_6x6_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/SwinIR_c180_6x6_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /paper_options/UNet_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/UNet_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /paper_options/Uformer_MultiFuseOut_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/Uformer_MultiFuseOut_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /paper_options/Uformer_MultiScaleHGT_MultiFuseOut_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/Uformer_MultiScaleHGT_MultiFuseOut_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /paper_options/Uformer_MultiScaleHGT_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/Uformer_MultiScaleHGT_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /paper_options/Uformer_multivarall_zscore_option.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/paper_options/Uformer_multivarall_zscore_option.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/run.sh -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/setup.cfg -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/x7zhong/TransfomerDownscaling/HEAD/train.py --------------------------------------------------------------------------------