Keras
vit
roadwork-miner / config.json
MichaelEdward's picture
Upload folder using huggingface_hub
f99a095 verified
raw
history blame contribute delete
545 Bytes
{
"_name_or_path": "natix-network-org/roadwork",
"architectures": ["ViTForImageClassification"],
"id2label": { "0": "None", "1": "Roadwork" },
"label2id": { "None": 0, "Roadwork": 1 },
"image_size": 224,
"model_type": "vit",
"num_labels": 2,
"problem_type": "single_label_classification",
"task": "image-classification",
"combined_pipeline": true,
"pipeline_components": [
"ViT (from natix-network-org/roadwork)",
"YOLO (best.pt in this repo)",
"EfficientNetV2 + fusion (final_output.keras in this repo)"
]
}