File size: 1,806 Bytes
7667a87 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 | import numpy as np
import os
SEVIR_DATA_TYPES = ['vis', 'ir069', 'ir107', 'vil', 'lght']
SEVIR_RAW_DTYPES = {
'vis': np.int16,
'ir069': np.int16,
'ir107': np.int16,
'vil': np.uint8,
'lght': np.int16
}
LIGHTING_FRAME_TIMES = np.arange(- 120.0, 125.0, 5) * 60
SEVIR_DATA_SHAPE = {
'lght': (48, 48)
}
PREPROCESS_SCALE_SEVIR = {
'vis': 1, # Not utilized in original paper
'ir069': 1 / 1174.68,
'ir107': 1 / 2562.43,
'vil': 1 / 47.54,
'lght': 1 / 0.60517
}
PREPROCESS_OFFSET_SEVIR = {
'vis': 0, # Not utilized in original paper
'ir069': 3683.58,
'ir107': 1552.80,
'vil': - 33.44,
'lght': - 0.02990
}
PREPROCESS_SCALE_01 = {
'vis': 1,
'ir069': 1,
'ir107': 1,
'vil': 1 / 255, # currently the only one implemented
'lght': 1
}
PREPROCESS_OFFSET_01 = {
'vis': 0,
'ir069': 0,
'ir107': 0,
'vil': 0, # currently the only one implemented
'lght': 0
}
# TODO: UPdate this if the code move elsewhere
SEVIR_ROOT_DIR = '/data/weather2025/sevir_full'
SEVIR_ROOT_LR_DIR = '/data/weather2025/sevir_lr'
# sevir
SEVIR_CATALOG = os.path.join(SEVIR_ROOT_DIR, "CATALOG.csv")
SEVIR_DATA_DIR = os.path.join(SEVIR_ROOT_DIR, "data")
SEVIR_RAW_SEQ_LEN = 49
SEVIR_INTERVAL_REAL_TIME = 5
SEVIR_H_W_SIZE = (384,384)
# sevir-lr
SEVIR_LR_CATALOG = os.path.join(SEVIR_ROOT_LR_DIR, "CATALOG.csv")
SEVIR_LR_DATA_DIR = os.path.join(SEVIR_ROOT_LR_DIR, "data")
SEVIR_LR_RAW_SEQ_LEN = 25
SEVIR_LR_INTERVAL_REAL_TIME = 10
SEVIR_LR_H_W_SIZE = (128,128)
# filtered_sevir_lr
SEVIR_LR_FILTERED_CATALOG = os.path.join(SEVIR_ROOT_LR_DIR, "CATALOG_ALT_VIL_VIS.csv")
# region Constraints
SAMPLE_LIST = ['random', 'sequent']
LAYOUT_LIST = ('NHWT', 'NTHW', 'NTCHW', 'NTHWC', 'TNHW', 'TNCHW')
SPLIT_MODE_LIST = ('ceil', 'floor', 'uneven')
# endregion |