{ | |
"_name_or_path": "/kaggle/working/saved_model_epoch19", | |
"architectures": [ | |
"MaskFormerForInstanceSegmentation" | |
], | |
"backbone_config": { | |
"depths": [ | |
2, | |
2, | |
18, | |
2 | |
], | |
"drop_path_rate": 0.3, | |
"embed_dim": 192, | |
"hidden_size": 1536, | |
"image_size": 384, | |
"in_channels": 3, | |
"model_type": "swin", | |
"num_heads": [ | |
6, | |
12, | |
24, | |
48 | |
], | |
"out_features": [ | |
"stage4" | |
], | |
"out_indices": [ | |
4 | |
], | |
"path_norm": true, | |
"pretrain_img_size": 384, | |
"window_size": 12 | |
}, | |
"ce_weight": 1.0, | |
"cross_entropy_weight": 1.0, | |
"decoder_config": { | |
"max_position_embeddings": 1024, | |
"model_type": "detr", | |
"scale_embedding": false | |
}, | |
"dice_weight": 1.0, | |
"fpn_feature_size": 256, | |
"id2label": { | |
"0": "water", | |
"1": "whitewater", | |
"2": "sand", | |
"3": "rocky_terrain", | |
"4": "agricultural", | |
"5": "vegetation", | |
"6": "road", | |
"7": "development", | |
"8": "other_natural_terrain", | |
"9": "unknown" | |
}, | |
"init_std": 0.02, | |
"init_xavier_std": 1.0, | |
"label2id": { | |
"agricultural": 4, | |
"development": 7, | |
"other_natural_terrain": 8, | |
"road": 6, | |
"rocky_terrain": 3, | |
"sand": 2, | |
"unknown": 9, | |
"vegetation": 5, | |
"water": 0, | |
"whitewater": 1 | |
}, | |
"mask_feature_size": 256, | |
"mask_weight": 20.0, | |
"model_type": "maskformer", | |
"no_object_weight": 0.1, | |
"num_attention_heads": 8, | |
"num_hidden_layers": 6, | |
"num_queries": 100, | |
"output_auxiliary_logits": null, | |
"torch_dtype": "float32", | |
"transformers_version": "4.35.0", | |
"use_auxiliary_loss": false | |
} | |