Tan Zi Xu commited on
Commit
de4e6b0
·
1 Parent(s): 623bc3b

classification al

Browse files
models/convNext_tiny_v2_traffic_lights/class_map.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "id": 0,
4
+ "name": "car"
5
+ },
6
+ {
7
+ "id": 1,
8
+ "name": "truck"
9
+ },
10
+ {
11
+ "id": 2,
12
+ "name": "bus"
13
+ },
14
+ {
15
+ "id": 3,
16
+ "name": "trailer"
17
+ },
18
+ {
19
+ "id": 4,
20
+ "name": "construction_vehicle"
21
+ },
22
+ {
23
+ "id": 5,
24
+ "name": "pedestrian"
25
+ },
26
+ {
27
+ "id": 6,
28
+ "name": "motorcycle"
29
+ },
30
+ {
31
+ "id": 7,
32
+ "name": "bicycle"
33
+ },
34
+ {
35
+ "id": 8,
36
+ "name": "traffic_cone"
37
+ },
38
+ {
39
+ "id": 9,
40
+ "name": "barrier"
41
+ },
42
+ {
43
+ "id": 10,
44
+ "name": "traffic_light"
45
+ }
46
+ ]
models/convNext_tiny_v2_traffic_lights/model_config.json ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "paths": {
3
+ "proj_root": "/home/shoo/personal/dlDev/school/classificationProject",
4
+ "meta_dir": "/home/shoo/personal/dlDev/school/classificationProject/meta",
5
+ "crops_dir": "/home/shoo/personal/dlDev/school/classificationProject",
6
+ "manifest": "/home/shoo/personal/dlDev/school/classificationProject/meta/classification/manifest_inc.csv",
7
+ "classes": "/home/shoo/personal/dlDev/school/classificationProject/meta/classes.json",
8
+ "runs_dir": "/home/shoo/personal/dlDev/school/classificationProject/runs"
9
+ },
10
+ "class_spec": {
11
+ "append": [
12
+ "traffic_light"
13
+ ]
14
+ },
15
+ "split": {
16
+ "train": 0.7,
17
+ "val": 0.15,
18
+ "test": 0.15,
19
+ "seed": 42
20
+ },
21
+ "filters": {
22
+ "min_iou": 0.6,
23
+ "min_ioa": 0.6
24
+ },
25
+ "dataloader": {
26
+ "batch_size": 64,
27
+ "num_workers": 6,
28
+ "pin_memory": true,
29
+ "persistent_workers": false,
30
+ "prefetch_factor": 2,
31
+ "drop_last": true
32
+ },
33
+ "preprocess": {
34
+ "image_size": 224,
35
+ "normalize": "imagenet",
36
+ "letterbox": true
37
+ },
38
+ "augment": {
39
+ "hflip_p": 0.5,
40
+ "vflip_p": 0.5,
41
+ "rotation_deg": 36,
42
+ "scale_min": 0.8,
43
+ "scale_max": 1.2
44
+ },
45
+ "model": {
46
+ "arch": "convnext_tiny",
47
+ "pretrained": true,
48
+ "head_hidden": [
49
+ 1024,
50
+ 512,
51
+ 256
52
+ ],
53
+ "head_dropout": 0.25,
54
+ "head_norm": "layernorm"
55
+ },
56
+ "stage_A": {
57
+ "epochs": 5,
58
+ "freeze": "all_but_head",
59
+ "optimizer": {
60
+ "type": "adamw",
61
+ "lr": 0.0005,
62
+ "weight_decay": 0.01
63
+ },
64
+ "scheduler": {
65
+ "type": "reduce_on_plateau",
66
+ "patience": 5,
67
+ "factor": 0.5,
68
+ "min_lr": 1e-06
69
+ },
70
+ "loss": {
71
+ "type": "cross_entropy",
72
+ "label_smoothing": 0.0,
73
+ "class_weights": {
74
+ "mode": "inv_freq",
75
+ "normalize_mean_to_1": true,
76
+ "clip_min": 0.25,
77
+ "clip_max": 4.0
78
+ }
79
+ }
80
+ },
81
+ "stage_B": {
82
+ "epochs": 95,
83
+ "unfreeze_patterns": [
84
+ "^features\\.7\\b",
85
+ "classifier"
86
+ ],
87
+ "optimizer": {
88
+ "type": "adamw",
89
+ "param_groups": [
90
+ {
91
+ "pattern": "^features\\.7\\b",
92
+ "lr": 0.0001,
93
+ "weight_decay": 0.01
94
+ },
95
+ {
96
+ "pattern": "classifier",
97
+ "lr": 0.0005,
98
+ "weight_decay": 0.01
99
+ }
100
+ ]
101
+ },
102
+ "scheduler": {
103
+ "type": "reduce_on_plateau",
104
+ "patience": 3,
105
+ "factor": 0.5,
106
+ "min_lr": 1e-06
107
+ },
108
+ "loss": {
109
+ "type": "focal",
110
+ "gamma": 2.0,
111
+ "alpha": {
112
+ "mode": "effective_number",
113
+ "beta": 0.999,
114
+ "normalize_mean_to_1": true
115
+ }
116
+ },
117
+ "last_k_blocks": 1
118
+ },
119
+ "metrics": {
120
+ "select_by": "val_macro_f1",
121
+ "early_stop_on": "none",
122
+ "early_stop_patience": 8
123
+ },
124
+ "logging": {
125
+ "save_confusion": true,
126
+ "save_curves": true,
127
+ "gradcam_samples_per_class": 0,
128
+ "print_stageB_trainable": true
129
+ }
130
+ }
models/manifest.json CHANGED
@@ -8,6 +8,11 @@
8
  "_comment": "Using 20251006_230926__convnext_tiny__exp02",
9
  "repo_id": "AAI3001-P20/streetview-convnext-tiny",
10
  "filename": "best.pth"
 
 
 
 
 
11
  },
12
  "detectors": {
13
  "yolo11n_v1": {
 
8
  "_comment": "Using 20251006_230926__convnext_tiny__exp02",
9
  "repo_id": "AAI3001-P20/streetview-convnext-tiny",
10
  "filename": "best.pth"
11
+ },
12
+ "convNext_tiny_v2_traffic_lights": {
13
+ "_comment": "Using 20251122_154701__convnext_tiny__cls_inc_kd_v1",
14
+ "repo_id": "AAI3001-P20/streetview-convnext-tiny-traffic-lights",
15
+ "filename": "best.pth"
16
  },
17
  "detectors": {
18
  "yolo11n_v1": {