Tasfiya025's picture
Create config.json
bb226c7 verified
{
"_name_or_path": "custom-sar-cnn-classifier",
"architectures": [
"ResNetForImageClassification"
],
"model_type": "image_classification",
"num_labels": 5,
"in_channels": 2,
"image_size": [128, 128],
"hidden_size": 2048,
"id2label": {
"0": "Water_Bodies",
"1": "Urban_Areas",
"2": "Forests",
"3": "Agriculture",
"4": "Barren_Land"
},
"label2id": {
"Water_Bodies": 0,
"Urban_Areas": 1,
"Forests": 2,
"Agriculture": 3,
"Barren_Land": 4
},
"problem_type": "single_label_classification",
"transformers_version": "4.36.0",
"base_model": "resnet50"
}