config.json

{
 "_name_or_path": "segformer-b0-CDL-S2-May19-v1-outputs/checkpoint-119450",
 "architectures": [
 "SegformerForSemanticSegmentation"
 ],
 "attention_probs_dropout_prob": 0.0,
 "classifier_dropout_prob": 0.1,
 "decoder_hidden_size": 256,
 "depths": [
 2,
 2,
 2,
 2
 ],
 "downsampling_rates": [
 1,
 4,
 8,
 16
 ],
 "drop_path_rate": 0.1,
 "hidden_act": "gelu",
 "hidden_dropout_prob": 0.0,
 "hidden_sizes": [
 32,
 64,
 160,
 256
 ],
 "id2label": {
 "0": "background",
 "1": "water",
 "2": "developed",
 "3": "corn",
 "4": "soybeans",
 "5": "wheat",
 "6": "other agriculture",
 "7": "forest/wetlands",
 "8": "open lands",
 "9": "barren"
 },
 "image_size": 224,
 "initializer_range": 0.02,
 "label2id": {
 "background": 0,
 "barren": 9,
 "corn": 3,
 "developed": 2,
 "forest/wetlands": 7,
 "open lands": 8,
 "other agriculture": 6,
 "soybeans": 4,
 "water": 1,
 "wheat": 5
 },
 "layer_norm_eps": 1e-06,
 "mlp_ratios": [
 4,
 4,
 4,
 4
 ],
 "model_type": "segformer",
 "num_attention_heads": [
 1,
 2,
 5,
 8
 ],
 "num_channels": 3,
 "num_encoder_blocks": 4,
 "patch_sizes": [
 7,
 3,
 3,
 3
 ],
 "reshape_last_stage": true,
 "semantic_loss_ignore_index": 0,
 "sr_ratios": [
 8,
 4,
 2,
 1
 ],
 "strides": [
 4,
 2,
 2,
 2
 ],
 "torch_dtype": "float32",
 "transformers_version": "4.37.0.dev0"
}