Spaces:
Runtime error
Runtime error
import os | |
CHARPOINT_CONVERSION_DICT = { | |
"": "leeg", | |
"101_Q19_2": "buitenkruin", | |
"101_Q19_3": "binnenkruin", | |
"101_Q19_5": "binnenteen", | |
"105_T09_11": "insteek_sloot", | |
"811_T13_8": "leeg", | |
"351_T03_10": "leeg", | |
"_T01_KKW": "leeg", | |
"108_Q06_250": "leeg", | |
"303_Q05_1": "leeg", | |
"353__11": "leeg", | |
"_T00_17": "leeg", | |
"109_Q08_13": "leeg", | |
"_Q07_KDM": "leeg", | |
"_Q07_KDW": "leeg", | |
'0': "leeg", | |
None: "leeg", | |
'nan': "leeg" | |
} | |
CLASS_DICT_REGIONAL = { | |
"leeg": 0, | |
"startpunt": 1, | |
"buitenkruin": 2, | |
"binnenkruin": 3, | |
"binnenteen": 4, | |
"insteek_sloot": 5 | |
} | |
WEIGHT_DICT_REGIONAL = [0.1, 1.0, 1.1, 1.0, 0.1] | |
CLASS_DICT_FULL = { | |
'leeg': 0, | |
'Maaiveld binnenwaarts': 1, | |
'Insteek sloot polderzijde': 2, | |
'Slootbodem polderzijde': 3, | |
'Slootbodem dijkzijde': 4, | |
'Insteek sloot dijkzijde': 5, | |
'Teen dijk binnenwaarts': 6, | |
'Kruin binnenberm': 7, | |
'Insteek binnenberm': 8, | |
'Kruin binnentalud': 9, | |
'Verkeersbelasting kant binnenwaarts': 9, # 10 | |
'Verkeersbelasting kant buitenwaarts': 10, | |
'Kruin buitentalud': 10, # 12 | |
'Insteek buitenberm': 11, | |
'Kruin buitenberm': 12, | |
'Teen dijk buitenwaarts': 13, | |
'Insteek geul': 14, | |
'Teen geul': 15, | |
'Maaiveld buitenwaarts': 16, | |
} | |
# TODO: write this out explicitely | |
WEIGHT_DICT_FULL = [1.0] * 17 | |
CLASS_DICT_SIMPLE = { | |
'leeg': 0, | |
'Maaiveld buitenwaarts': 1, | |
'Teen dijk buitenwaarts': 2, | |
'Kruin buitentalud': 3, | |
'Kruin binnentalud': 4, | |
'Teen dijk binnenwaarts': 5, | |
} | |
WEIGHT_DICT_SIMPLE = [0.1, 0.5, 0.7, 1.0, 1.0, 0.5] | |
CLASS_DICT_SIMPLE_SLOOT = { | |
'leeg': 0, | |
'Maaiveld buitenwaarts': 1, | |
'Teen dijk buitenwaarts': 2, | |
'Kruin buitentalud': 3, | |
'Kruin binnentalud': 4, | |
'Teen dijk binnenwaarts': 5, | |
'Insteek sloot dijkzijde': 6, | |
'Insteek sloot polderzijde': 7, | |
'Slootbodem polderzijde': 8, | |
'Slootbodem dijkzijde': 9, | |
} | |
WEIGHT_DICT_SIMPLE_SLOOT = [0.1, 0.1, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 0.1] | |
CLASS_DICT_SIMPLE_BERM = { | |
'leeg': 0, | |
'Maaiveld buitenwaarts': 1, | |
'Teen dijk buitenwaarts': 2, | |
'Kruin buitentalud': 3, | |
'Kruin binnentalud': 4, | |
'Teen dijk binnenwaarts': 5, | |
'Insteek sloot dijkzijde': 6, | |
'Insteek sloot polderzijde': 7, | |
'Slootbodem polderzijde': 8, | |
'Slootbodem dijkzijde': 9, | |
'Kruin binnenberm': 10, | |
'Insteek binnenberm': 11, | |
} | |
WEIGHT_DICT_SIMPLE_BERM = [0.1, 0.1, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 0.1] | |
HEADER = ["LOCATIONID", | |
"X_Maaiveld binnenwaarts", | |
"Y_Maaiveld binnenwaarts", | |
"Z_Maaiveld binnenwaarts", | |
"X_Insteek sloot polderzijde", | |
"Y_Insteek sloot polderzijde", | |
"Z_Insteek sloot polderzijde", | |
"X_Slootbodem polderzijde", | |
"Y_Slootbodem polderzijde", | |
"Z_Slootbodem polderzijde", | |
"X_Slootbodem dijkzijde", | |
"Y_Slootbodem dijkzijde", | |
"Z_Slootbodem dijkzijde", | |
"X_Insteek sloot dijkzijde", | |
"Y_Insteek sloot dijkzijde", | |
"Z_Insteek sloot dijkzijde", | |
"X_Teen dijk binnenwaarts", | |
"Y_Teen dijk binnenwaarts", | |
"Z_Teen dijk binnenwaarts", | |
"X_Kruin binnenberm", | |
"Y_Kruin binnenberm", | |
"Z_Kruin binnenberm", | |
"X_Insteek binnenberm", | |
"Y_Insteek binnenberm", | |
"Z_Insteek binnenberm", | |
"X_Kruin binnentalud", | |
"Y_Kruin binnentalud", | |
"Z_Kruin binnentalud", | |
"X_Verkeersbelasting kant binnenwaarts", | |
"Y_Verkeersbelasting kant binnenwaarts", | |
"Z_Verkeersbelasting kant binnenwaarts", | |
"X_Verkeersbelasting kant buitenwaarts", | |
"Y_Verkeersbelasting kant buitenwaarts", | |
"Z_Verkeersbelasting kant buitenwaarts", | |
"X_Kruin buitentalud", | |
"Y_Kruin buitentalud", | |
"Z_Kruin buitentalud", | |
"X_Insteek buitenberm", | |
"Y_Insteek buitenberm", | |
"Z_Insteek buitenberm", | |
"X_Kruin buitenberm", | |
"Y_Kruin buitenberm", | |
"Z_Kruin buitenberm", | |
"X_Teen dijk buitenwaarts", | |
"Y_Teen dijk buitenwaarts", | |
"Z_Teen dijk buitenwaarts", | |
"X_Insteek geul", | |
"Y_Insteek geul", | |
"Z_Insteek geul", | |
"X_Teen geul", | |
"Y_Teen geul", | |
"Z_Teen geul", | |
"X_Maaiveld buitenwaarts", | |
"Y_Maaiveld buitenwaarts", | |
"Z_Maaiveld buitenwaarts"] | |
SCALER_PATH = os.path.join("data", "trained_models", "scaler.pik") | |
MODEL_PATH = os.path.join('data', 'trained_models', 'dijknet_simple_95.pt') | |
INVERSE_CLASS_DICT_FULL = {v: k for k, v in CLASS_DICT_FULL.items()} | |
INVERSE_CLASS_DICT_SIMPLE = {v: k for k, v in CLASS_DICT_SIMPLE.items()} | |
INVERSE_CLASS_DICT_SIMPLE_BERM = {v: k for k, v in CLASS_DICT_SIMPLE_BERM.items()} | |
INVERSE_CLASS_DICT_SIMPLE_SLOOT = {v: k for k, v in CLASS_DICT_SIMPLE_SLOOT.items()} | |
INVERSE_CLASS_DICT_REGIONAL = {v: k for k, v in CLASS_DICT_REGIONAL.items()} | |
# manual mappings to get the correct names for plotting later | |
if 11 in INVERSE_CLASS_DICT_FULL: | |
INVERSE_CLASS_DICT_FULL[10] = 'Kruin buitentalud' | |