|
# proto-file: deeplab2/config.proto |
|
# proto-message: ExperimentOptions |
|
|
|
model_options { |
|
decoder { |
|
feature_key: "feature_semantic" |
|
atrous_rates: 6 |
|
atrous_rates: 12 |
|
atrous_rates: 18 |
|
} |
|
|
|
backbone { |
|
name: "max_deeplab_s" |
|
output_stride: 16 |
|
} |
|
|
|
max_deeplab { |
|
pixel_space_head { |
|
output_channels: 128 |
|
head_channels: 256 |
|
} |
|
auxiliary_low_level { |
|
feature_key: "res3" |
|
channels_project: 64 |
|
} |
|
auxiliary_low_level { |
|
feature_key: "res2" |
|
channels_project: 32 |
|
} |
|
auxiliary_semantic_head { |
|
# Example for COCO. |
|
output_channels: 134 |
|
head_channels: 256 |
|
} |
|
} |
|
} |
|
|
|
train_dataset_options { |
|
crop_size: 65 |
|
crop_size: 65 |
|
} |
|
|