Spaces:
Runtime error
Runtime error
File size: 5,281 Bytes
b9bac12 |
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 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 |
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'
|