-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
62 lines (58 loc) · 1.28 KB
/
config.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
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
# Config file
paths:
metrics: './results/logs'
figures: './results/figures'
models: './results/models'
model:
type: ['pspnet', 'deeplabv3_plus', 'unet']
backbone: resnet101
pretrained: imagenet
classes: 3
activation: null
training:
epochs: 2
loss_function:
type: focal_loss
metrics:
- foreground_acc
- DiceMulti
- JaccardCoeffMulti
data:
path_to_dataset: './data/processed/train'
path_test_dataset: './data/processed/val'
batch_size: 32
validation_split: 0.1
augmentation:
resize: [8, 8]
shadow_transform:
num_shadows: 3
min_opacity: 0.5
max_opacity: 0.75
normalize: imagenet_stats
aug_transforms:
mult: 2
do_flip: true
flip_vert: true
max_rotate: 90.0
min_zoom: 0.8
max_zoom: 3.0
max_lighting: 0.2
max_warp: 0.2
p_affine: 0.75
p_lighting: 0.75
mode: bilinear
pad_mode: reflection
align_corners: true
batch: false
min_scale: 1
mapping_class_color:
- color: [107, 142, 35] # No se cual era
class: 0
- color: [102, 102, 156] # No se cual era
class: 0
- color: [128, 64, 128] # Road
class: 1
- color: [0, 0, 142] # Car
class: 2
- color: [0, 0, 0] # Background
class: 0