alsatix_image_control_v2 / model_config.json
iammbrn's picture
Update model_config.json
c104af3 verified
raw
history blame contribute delete
411 Bytes
{
"model_type": "resnet",
"num_labels": 5,
"id2label": {
"0": "alkol",
"1": "normal",
"2": "nsfw",
"3": "silah",
"4": "tutun"
},
"label2id": {
"alkol": 0,
"normal": 1,
"nsfw": 2,
"silah": 3,
"tutun": 4
},
"input_size": 224,
"num_classes": 5,
"pretrained": true,
"architecture": "ResNet50"
}