-
Notifications
You must be signed in to change notification settings - Fork 117
/
config.1_FRLR.unetxst.yml
30 lines (25 loc) · 1.25 KB
/
config.1_FRLR.unetxst.yml
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
input-training: [../data/1_FRLR/train/front, ../data/1_FRLR/train/rear, ../data/1_FRLR/train/left, ../data/1_FRLR/train/right]
label-training: ../data/1_FRLR/train/bev+occlusion
max-samples-training: 100000
input-validation: [../data/1_FRLR/val/front, ../data/1_FRLR/val/rear, ../data/1_FRLR/val/left, ../data/1_FRLR/val/right]
label-validation: ../data/1_FRLR/val/bev+occlusion
max-samples-validation: 10000
image-shape: [256, 512]
one-hot-palette-input: one_hot_conversion/convert_10.xml
one-hot-palette-label: one_hot_conversion/convert_9+occl.xml
model: architecture/uNetXST.py
unetxst-homographies: ../preprocessing/homography_converter/uNetXST_homographies/1_FRLR.py
epochs: 100
batch-size: 5
learning-rate: 1e-4
loss-weights: [0.98684351, 2.2481491, 10.47452063, 4.78351389, 7.01028204, 8.41360361, 10.91633349, 2.38571558, 1.02473193, 2.79359197]
early-stopping-patience: 20
save-interval: 5
output-dir: output
# for training continuation, evaluation and prediction only
class-names: [road, sidewalk, person, car, truck, bus, bike, obstacle, vegetation, occluded]
# model-weights:
# for predict.py only
input-testing: [../data/1_FRLR/val/front, ../data/1_FRLR/val/rear, ../data/1_FRLR/val/left, ../data/1_FRLR/val/right]
max-samples-testing: 10000
# prediction-dir: