camstack-models / vehicleClassification /vehicle-type-v1 /vehicle-type-v1-labels.json
apocaliss92's picture
add vehicleClassification/vehicle-type-v1 small classifier (onnx+openvino+coreml+labels)
604e912 verified
Raw
History Blame Contribute Delete
641 Bytes
{
"labels": [
"Ambulance",
"Bicycle",
"Bus",
"Car",
"Motorcycle",
"Taxi",
"Truck",
"Van"
],
"input_size": 224,
"num_classes": 8,
"preprocess": {
"normalize": "imagenet",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"channels": "RGB",
"layout": "NCHW"
},
"arch": "mobilenet_v3_large",
"source": "torchvision IMAGENET1K_V2 fine-tuned on DrBimmer/vehicle-classification road subset",
"license": "BSD-3-Clause (torchvision) / dataset per-source",
"output": "logits (apply softmax)",
"best_val_acc": 0.8938
}