Vansh180 commited on
Commit
75b2867
·
verified ·
1 Parent(s): b1f4c16

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +75 -0
config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "convnext_base.fb_in22k_ft_in1k_384",
3
+ "num_classes": 6,
4
+ "class_names": [
5
+ "HDPE",
6
+ "LDPE",
7
+ "Other",
8
+ "PET",
9
+ "PP",
10
+ "PS"
11
+ ],
12
+ "class_to_idx": {
13
+ "HDPE": 0,
14
+ "LDPE": 1,
15
+ "Other": 2,
16
+ "PET": 3,
17
+ "PP": 4,
18
+ "PS": 5
19
+ },
20
+ "idx_to_class": {
21
+ "0": "HDPE",
22
+ "1": "LDPE",
23
+ "2": "Other",
24
+ "3": "PET",
25
+ "4": "PP",
26
+ "5": "PS"
27
+ },
28
+ "img_size": 384,
29
+ "imagenet_mean": [
30
+ 0.485,
31
+ 0.456,
32
+ 0.406
33
+ ],
34
+ "imagenet_std": [
35
+ 0.229,
36
+ 0.224,
37
+ 0.225
38
+ ],
39
+ "hierarchy": {
40
+ "PET": {
41
+ "category": "Rigid Thermoplastic",
42
+ "recyclable": true,
43
+ "code": 1
44
+ },
45
+ "HDPE": {
46
+ "category": "Rigid Thermoplastic",
47
+ "recyclable": true,
48
+ "code": 2
49
+ },
50
+ "PP": {
51
+ "category": "Rigid Thermoplastic",
52
+ "recyclable": true,
53
+ "code": 5
54
+ },
55
+ "LDPE": {
56
+ "category": "Flexible Thermoplastic",
57
+ "recyclable": true,
58
+ "code": 4
59
+ },
60
+ "PS": {
61
+ "category": "Brittle Thermoplastic",
62
+ "recyclable": false,
63
+ "code": 6
64
+ },
65
+ "Other": {
66
+ "category": "Unknown",
67
+ "recyclable": null,
68
+ "code": 7
69
+ }
70
+ },
71
+ "test_accuracy": 0.7801980198019802,
72
+ "test_macro_f1": 0.8077352266209736,
73
+ "best_val_f1": 0.7676573659411366,
74
+ "strategy": "Direct classification (no YOLO), full LLRD, EMA, 5-view TTA"
75
+ }