File size: 667 Bytes
50318d8 |
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 |
model:
image_encoder:
backbone:
pretrained: true
frozen: true
output_dim: 128
name: feature_extractor_DPT
segmentation_head:
dropout_rate: 0.2
name: map_perception_net
num_classes: 6
latent_dim: 128
z_max: 50
x_max: 25
pixel_per_meter: 2
num_scale_bins: 32
loss:
num_classes: 6
xent_weight: 1.0
dice_weight: 1.0
focal_loss: false
focal_loss_gamma: 2.0
requires_frustrum: true
requires_flood_mask: false
class_weights:
- 1.00351229
- 4.34782609
- 1.00110121
- 1.03124678
- 6.69792364
- 7.55857899
label_smoothing: 0.1
scale_range:
- 0
- 9
z_min: null |