| from data import dutils | |
| SEVIR_13_12 = { | |
| 'meta': { | |
| 'dataset': 'SEVIR', | |
| 'seq_len': 13, | |
| 'out_len': 12, | |
| 'metrics': ['mae', 'mse', 'ssim', 'psnr', 'lpips64', | |
| 'csi-16', 'csi-74', 'csi-133', 'csi-160', 'csi-181', 'csi-219', | |
| 'csi_4-16', 'csi_4-74', 'csi_4-133', 'csi_4-160', 'csi_4-181', 'csi_4-219', | |
| 'csi_16-16', 'csi_16-74', 'csi_16-133', 'csi_16-160', 'csi_16-181', 'csi_16-219', | |
| 'hss-16', 'hss-74', 'hss-133', 'hss-160', 'hss-181', 'hss-219' ], | |
| }, | |
| 'param': { | |
| 'seq_len': 25, | |
| 'data_types': ['vil'], | |
| 'sample_mode': 'sequent', | |
| 'layout': 'NTCHW', | |
| 'raw_seq_len': 25, | |
| 'start_date': dutils.SEVIR_TRAIN_TEST_SPLIT_DATE, | |
| 'end_date': None, | |
| }, | |
| 'savedir': 'sevir' | |
| } | |
| HKO7_5_20 = { | |
| 'meta': { | |
| 'dataset': 'HKO-7', | |
| 'seq_len': 5, | |
| 'out_len': 20, | |
| 'metrics': ['mae', 'mse', 'ssim', 'psnr', 'lpips64', | |
| 'csi-84', 'csi-117', 'csi-140', 'csi-158', 'csi-185', | |
| 'csi_4-84', 'csi_4-117', 'csi_4-140', 'csi_4-158', 'csi_4-185', | |
| 'csi_16-84', 'csi_16-117', 'csi_16-140', 'csi_16-158', 'csi_16-185', | |
| 'hss-84', 'hss-117', 'hss-140', 'hss-158', 'hss-185'], | |
| }, | |
| 'param': { | |
| 'pd_path': 'data/HKO-7/samplers/hko7_cloudy_days_t20_test.txt.pkl', | |
| 'sample_mode': 'sequent', | |
| 'seq_len': 25, | |
| 'stride': 13, | |
| }, | |
| 'savedir': 'hko-7' | |
| } | |
| METEONET_5_20 = { | |
| 'meta': { | |
| 'dataset': 'meteonet', | |
| 'seq_len': 5, | |
| 'out_len': 20, | |
| 'metrics': ['mae', 'mse', 'ssim', 'psnr', 'lpips64', | |
| 'csi-44', 'csi-64', 'csi-87', 'csi-117', | |
| 'csi_4-44', 'csi_4-64', 'csi_4-87', 'csi_4-117', | |
| 'csi_16-44', 'csi_16-64', 'csi_16-87', 'csi_16-117', | |
| 'hss-44', 'hss-64', 'hss-87', 'hss-117'] | |
| }, | |
| 'param': { | |
| 'img_size': 128, | |
| 'in_len': 5, | |
| }, | |
| 'savedir': 'meteonet' | |
| } |