config.yaml

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