Hugging Face's logo

Danieli1021
/
streetvision

Safetensors
custom-resnet-binary-classifier
Model card Files
xet
 Community
Danieli1021 commited on
Commit
b356704
·
verified ·
1 Parent(s): 2d4c523

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +21 -21
config.json CHANGED
@@ -1,21 +1,21 @@
1
- {
2
- "architectures": [
3
- "RoadworkBinaryClassifier"
4
- ],
5
- "model_type": "custom-resnet-binary-classifier",
6
- "backbone_name": "resnet50",
7
- "num_labels": 1,
8
- "id2label": {
9
- "0": "non-roadwork",
10
- "1": "roadwork"
11
- },
12
- "label2id": {
13
- "non-roadwork": 0,
14
- "roadwork": 1
15
- },
16
- "problem_type": "binary_classification",
17
- "image_size": 320,
18
- "torch_dtype": "float32",
19
- "best_threshold": 0.3,
20
- "best_val_f1": 0.9875060067275349
21
- }
 
1
+ {
2
+ "architectures": [
3
+ "RoadworkBinaryClassifier"
4
+ ],
5
+ "model_type": "custom-resnet-binary-classifier",
6
+ "backbone_name": "resnet50",
7
+ "num_labels": 1,
8
+ "id2label": {
9
+ "0": "None",
10
+ "1": "Roadwork"
11
+ },
12
+ "label2id": {
13
+ "None": 0,
14
+ "Roadwork": 1
15
+ },
16
+ "problem_type": "binary_classification",
17
+ "image_size": 320,
18
+ "torch_dtype": "float32",
19
+ "best_threshold": 0.3,
20
+ "best_val_f1": 0.9875060067275349
21
+ }