damage-classifier-multi-task / model_config.json
narinzar's picture
Upload damage classification model
6ac5a86 verified
raw
history blame contribute delete
426 Bytes
{
"classification_type": "multi_task",
"num_labels": null,
"model_name": "google/vit-base-patch16-224",
"item_categories": [
"microwave",
"wall",
"window",
"fence",
"glass",
"fishbowl"
],
"damage_types": [
"scratch",
"dent",
"break",
"burn",
"water_damage"
],
"damage_severity": [
"no_damage",
"minor_damage",
"moderate_damage",
"severe_damage"
]
}