streetvision / config.json
Danieli1021's picture
Update config.json
b356704 verified
{
"architectures": [
"RoadworkBinaryClassifier"
],
"model_type": "custom-resnet-binary-classifier",
"backbone_name": "resnet50",
"num_labels": 1,
"id2label": {
"0": "None",
"1": "Roadwork"
},
"label2id": {
"None": 0,
"Roadwork": 1
},
"problem_type": "binary_classification",
"image_size": 320,
"torch_dtype": "float32",
"best_threshold": 0.3,
"best_val_f1": 0.9875060067275349
}