disaster-prediction / config.json
Luwayy's picture
Create config.json
2ffd2b1
raw
history blame
1.04 kB
{
"model_type": "resnet",
"model": "ResNet50",
"weights": "kaggle/working/disaster_model",
"license": "mit",
"language": "en",
"metrics": {
"main": "accuracy"
},
"pipeline_tag": "image-classification",
"classes": [
{
"id": 0,
"name": "Damaged_Infrastructure",
"height": 256,
"width": 256,
"scale_by": 1
},
{
"id": 1,
"name": "Fire_Disaster",
"height": 256,
"width": 256,
"scale_by": 1
},
{
"id": 2,
"name": "Human_Damage",
"height": 256,
"width": 256,
"scale_by": 1
},
{
"id": 3,
"name": "Land_Disaster",
"height": 256,
"width": 256,
"scale_by": 1
},
{
"id": 4,
"name": "Non_Damage",
"height": 256,
"width": 256,
"scale_by": 1
},
{
"id": 5,
"name": "Water_Disaster",
"height": 256,
"width": 256,
"scale_by": 1
}
],
"preprocessing": {
"resize": [256, 256],
"scale": 255
}
}