File size: 3,881 Bytes
e07eac0 f4ad4b4 |
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 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 |
name: "RDA_ResNet101_DeepLabv3plus_simple_unadjusted"
model: "DeepLabV3PlusResNet"
task: "RDA"
masking_strategy_parameters:
road_width_pixels: 40
train_location_strategy: "grid"
train_location_parameters:
sample_generator_process_pool_size: 6
adjustment_buffer_distance_px: 326
train_presentation_strategy: "weighted"
train_presentation_strategy_parameters:
length: 4000
expected_class_balances:
Not Able To Determine: 1
Partial Flooding: 1
Partial Obstruction: 1
Partial Road Condition: 1
Particulate Partial: 1
Particulate Total: 1
Road Line: 1
Total Destruction: 1
Total Flooding: 1
Total Obstruction: 1
balance_monitor: "pixel"
train_dataset_adaptor_parameters:
tile_x: 1024
tile_y: 1024
mask_x: 1024
mask_y: 1024
backend: "auto"
train_annotator_parameters:
generate_adjusted_samples: False
validation_location_strategy: "grid"
validation_location_parameters:
sample_generator_process_pool_size: 6
adjustment_buffer_distance_px: 326
validation_presentation_strategy: "indexed"
validation_dataset_adaptor_parameters:
tile_x: 1024
tile_y: 1024
mask_x: 1024
mask_y: 1024
backend: "auto"
validation_annotator_parameters:
generate_adjusted_samples: False
channel_maps:
background_class_idx: 0
default_label: Road Line
input_dataset_class_2_idx_map:
Background: 0
Not Able To Determine: 3
Partial Flooding: 2
Partial Obstruction: 2
Partial Road Condition: 2
Particulate Partial: 2
Particulate Total: 3
Road Line: 1
Total Destruction: 3
Total Flooding: 3
Total Obstruction: 3
model_class_2_color_map:
Background:
alpha: 0
blue: 0
green: 0
red: 0
Partial:
alpha: 255
blue: 0
green: 255
red: 255
Road Line:
alpha: 255
blue: 0
green: 255
red: 0
Total:
alpha: 255
blue: 0
green: 0
red: 255
output_class_2_idx_map:
Background: 0
Partial: 2
Road Line: 1
Total: 3
input:
road_line_buffer_width_pixels: 40
road_line_segment_length_pixels: 40
channels:
blue: 2
green: 1
mask: 3
red: 0
normalized_inputs: true
mask_input: false
sample_generator_parameters:
label_tolerance: 0
normalized_inputs: true
validation_parameters:
batch_size: 1
validation_monitor: "val_macro_iou"
validation_monitor_mode: "max"
validation_checkpoint_save_top_k: 3
validation_reduce_lr_on_plateau_patience: 3
validation_reduce_lr_on_plateau_cooldown: 3
training_parameters:
gradient_clip_val: 10.0
batch_size: 1
grad_accumulation: 32
l1_reg: 1.0e-06
l2_reg: 1.0e-05
log_images_every_n_steps: 100
mask_x: 1024
mask_y: 1024
max_epochs: 100
optimizer_parameters:
name: "AdamW"
lr: 0.0005
weight_decay: 0.00001
output_class_weights_strategy: "ipw"
normalize_weights: True #Added
output_class_weights:
Background: 1.0e-06
Partial: 45
Road Line: 1
Total: 35
samples_per_epoch: 4000
tile_x: 1024
tile_y: 1024
loss_parameters:
loss: [["focal", 1.0]]
gamma: 3
alpha: 0.25
bins: 10
momentum: 0.9
smooth: 1.0e-06
model_parameters:
n_cls: 4
encoder_parameters:
depth: 101
num_stages: 4
out_indices: [0, 1, 2, 3]
dilations: [1, 1, 2, 4]
strides: [1, 2, 1, 1]
norm_eval: False
contract_dilation: True
decoder_parameters:
in_channels: 2048
in_index: 3
channels: 512
dilations: [1, 12, 24, 36]
c1_in_channels: 256
c1_channels: 48
dropout_ratio: 0.1
align_corners: False
auxiliary_head_parameters:
in_channels: 1024
in_index: 2
channels: 256
num_convs: 1
concat_input: False
dropout_ratio: 0.1
align_corners: False |