nameplate-classifier / config.json
avewright's picture
Add nameplate classifier model
3f2d6ef verified
raw
history blame contribute delete
392 Bytes
{
"architectures": [
"LightweightNameplateClassifier"
],
"model_type": "image-classification",
"num_classes": 2,
"id2label": {
"0": "no_nameplate",
"1": "has_nameplate"
},
"label2id": {
"no_nameplate": 0,
"has_nameplate": 1
},
"image_size": 224,
"backbone": "mobilenet_v2",
"framework": "pytorch",
"task": "image-classification"
}