Mapper / config.yaml
Jiaye Zou
update: gradio app with docker
50318d8
raw
history blame
667 Bytes
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