Hugging Face's logo

jinpeng0528
/
STAR

Model card Files
xet
Community
jinpeng0528 commited on
Commit
ae767a8
·
verified ·
1 Parent(s): 6fa9758

Upload 6 files

Browse files
Files changed (6) hide show
  1. ade_overlapped_100-10_STAR/config.json +109 -0
  2. ade_overlapped_100-10_STAR/overlapped_100-10_STAR.pth +3 -0
  3. ade_overlapped_100-50_STAR/config.json +109 -0
  4. ade_overlapped_100-50_STAR/overlapped_100-50_STAR.pth +3 -0
  5. ade_overlapped_50-50_STAR/config.json +109 -0
  6. ade_overlapped_50-50_STAR/overlapped_50-50_STAR.pth +3 -0
ade_overlapped_100-10_STAR/config.json ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "STAR",
3
+ "n_gpu": 2,
4
+ "multiprocessing_distributed": true,
5
+ "rank": 0,
6
+ "world_size": 1,
7
+ "dist_backend": "nccl",
8
+ "dist_url": "tcp://127.0.0.1:12345",
9
+ "seed": 1,
10
+ "use_amp": true,
11
+ "arch": {
12
+ "type": "DeepLabV3",
13
+ "args": {
14
+ "output_stride": 16,
15
+ "norm_act": "bn_sync",
16
+ "backbone_pretrained": true,
17
+ "freeze_all_bn": true
18
+ }
19
+ },
20
+ "data_loader": {
21
+ "type": "ADEIncrementalDataLoader",
22
+ "args": {
23
+ "task": {
24
+ "name": "100-10",
25
+ "step": 5,
26
+ "setting": "overlap",
27
+ "idxs_path": "./data/"
28
+ },
29
+ "train": {
30
+ "args": {
31
+ "transform": true,
32
+ "transform_args": {
33
+ "crop_size": 512
34
+ },
35
+ "masking_value": 0
36
+ },
37
+ "batch_size": 12
38
+ },
39
+ "val": {
40
+ "cross_val": false,
41
+ "args": {
42
+ "transform": true,
43
+ "transform_args": {
44
+ "crop_size": 512
45
+ },
46
+ "masking_value": 255
47
+ },
48
+ "batch_size": 1
49
+ },
50
+ "test": {
51
+ "args": {
52
+ "transform": true,
53
+ "transform_args": {
54
+ "crop_size": 512
55
+ },
56
+ "masking_value": 255
57
+ },
58
+ "batch_size": 1
59
+ },
60
+ "num_workers": 4,
61
+ "pin_memory": true,
62
+ "memory": {
63
+ "mem_size": 0
64
+ }
65
+ }
66
+ },
67
+ "optimizer": {
68
+ "type": "SGD",
69
+ "args": {
70
+ "lr": 0.00025,
71
+ "weight_decay": 0.0001,
72
+ "momentum": 0.9,
73
+ "nesterov": true
74
+ }
75
+ },
76
+ "hyperparameter": {
77
+ "pos_weight": 30.0,
78
+ "mbce": 1,
79
+ "pkd": 5,
80
+ "cont": 0.05
81
+ },
82
+ "lr_scheduler": {
83
+ "type": "WarmupPolyLR",
84
+ "args": {
85
+ "warmup_iters": 0.2
86
+ }
87
+ },
88
+ "evaluator": {
89
+ "type": "Evaluator",
90
+ "args": {}
91
+ },
92
+ "metrics": [
93
+ "Pixel_Accuracy",
94
+ "Pixel_Accuracy_Class",
95
+ "Mean_Intersection_over_Union"
96
+ ],
97
+ "trainer": {
98
+ "epochs": 100,
99
+ "save_dir": "saved_ade",
100
+ "save_period": 10,
101
+ "validation_period": 10,
102
+ "verbosity": 2,
103
+ "monitor": "off",
104
+ "early_stop": -1,
105
+ "reset_best_mnt": false,
106
+ "tensorboard": false
107
+ },
108
+ "test": false
109
+ }
ade_overlapped_100-10_STAR/overlapped_100-10_STAR.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3dfeb32474147db73a9d217525167992c80cd821520ce96f5d7d51f197a8ea3b
3
+ size 469582911
ade_overlapped_100-50_STAR/config.json ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "STAR",
3
+ "n_gpu": 2,
4
+ "multiprocessing_distributed": true,
5
+ "rank": 0,
6
+ "world_size": 1,
7
+ "dist_backend": "nccl",
8
+ "dist_url": "tcp://127.0.0.1:12370",
9
+ "seed": 1,
10
+ "use_amp": true,
11
+ "arch": {
12
+ "type": "DeepLabV3",
13
+ "args": {
14
+ "output_stride": 16,
15
+ "norm_act": "bn_sync",
16
+ "backbone_pretrained": true,
17
+ "freeze_all_bn": true
18
+ }
19
+ },
20
+ "data_loader": {
21
+ "type": "ADEIncrementalDataLoader",
22
+ "args": {
23
+ "task": {
24
+ "name": "100-50",
25
+ "step": 1,
26
+ "setting": "overlap",
27
+ "idxs_path": "./data/"
28
+ },
29
+ "train": {
30
+ "args": {
31
+ "transform": true,
32
+ "transform_args": {
33
+ "crop_size": 512
34
+ },
35
+ "masking_value": 0
36
+ },
37
+ "batch_size": 12
38
+ },
39
+ "val": {
40
+ "cross_val": false,
41
+ "args": {
42
+ "transform": true,
43
+ "transform_args": {
44
+ "crop_size": 512
45
+ },
46
+ "masking_value": 255
47
+ },
48
+ "batch_size": 1
49
+ },
50
+ "test": {
51
+ "args": {
52
+ "transform": true,
53
+ "transform_args": {
54
+ "crop_size": 512
55
+ },
56
+ "masking_value": 255
57
+ },
58
+ "batch_size": 1
59
+ },
60
+ "num_workers": 4,
61
+ "pin_memory": true,
62
+ "memory": {
63
+ "mem_size": 0
64
+ }
65
+ }
66
+ },
67
+ "optimizer": {
68
+ "type": "SGD",
69
+ "args": {
70
+ "lr": 0.00025,
71
+ "weight_decay": 0.0001,
72
+ "momentum": 0.9,
73
+ "nesterov": true
74
+ }
75
+ },
76
+ "hyperparameter": {
77
+ "pos_weight": 30.0,
78
+ "mbce": 1,
79
+ "pkd": 5,
80
+ "cont": 0.05
81
+ },
82
+ "lr_scheduler": {
83
+ "type": "WarmupPolyLR",
84
+ "args": {
85
+ "warmup_iters": 0.2
86
+ }
87
+ },
88
+ "evaluator": {
89
+ "type": "Evaluator",
90
+ "args": {}
91
+ },
92
+ "metrics": [
93
+ "Pixel_Accuracy",
94
+ "Pixel_Accuracy_Class",
95
+ "Mean_Intersection_over_Union"
96
+ ],
97
+ "trainer": {
98
+ "epochs": 100,
99
+ "save_dir": "saved_ade",
100
+ "save_period": 10,
101
+ "validation_period": 10,
102
+ "verbosity": 2,
103
+ "monitor": "off",
104
+ "early_stop": -1,
105
+ "reset_best_mnt": false,
106
+ "tensorboard": false
107
+ },
108
+ "test": false
109
+ }
ade_overlapped_100-50_STAR/overlapped_100-50_STAR.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59b8e17aecc6b35f621afa22006cf6cab30ac126a5ffc955bd064bf24bc18aae
3
+ size 469578335
ade_overlapped_50-50_STAR/config.json ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "STAR",
3
+ "n_gpu": 2,
4
+ "multiprocessing_distributed": true,
5
+ "rank": 0,
6
+ "world_size": 1,
7
+ "dist_backend": "nccl",
8
+ "dist_url": "tcp://127.0.0.1:12345",
9
+ "seed": 1,
10
+ "use_amp": true,
11
+ "arch": {
12
+ "type": "DeepLabV3",
13
+ "args": {
14
+ "output_stride": 16,
15
+ "norm_act": "bn_sync",
16
+ "backbone_pretrained": true,
17
+ "freeze_all_bn": true
18
+ }
19
+ },
20
+ "data_loader": {
21
+ "type": "ADEIncrementalDataLoader",
22
+ "args": {
23
+ "task": {
24
+ "name": "50-50",
25
+ "step": 2,
26
+ "setting": "overlap",
27
+ "idxs_path": "./data/"
28
+ },
29
+ "train": {
30
+ "args": {
31
+ "transform": true,
32
+ "transform_args": {
33
+ "crop_size": 512
34
+ },
35
+ "masking_value": 0
36
+ },
37
+ "batch_size": 12
38
+ },
39
+ "val": {
40
+ "cross_val": false,
41
+ "args": {
42
+ "transform": true,
43
+ "transform_args": {
44
+ "crop_size": 512
45
+ },
46
+ "masking_value": 255
47
+ },
48
+ "batch_size": 1
49
+ },
50
+ "test": {
51
+ "args": {
52
+ "transform": true,
53
+ "transform_args": {
54
+ "crop_size": 512
55
+ },
56
+ "masking_value": 255
57
+ },
58
+ "batch_size": 1
59
+ },
60
+ "num_workers": 4,
61
+ "pin_memory": true,
62
+ "memory": {
63
+ "mem_size": 0
64
+ }
65
+ }
66
+ },
67
+ "optimizer": {
68
+ "type": "SGD",
69
+ "args": {
70
+ "lr": 0.00025,
71
+ "weight_decay": 0.0001,
72
+ "momentum": 0.9,
73
+ "nesterov": true
74
+ }
75
+ },
76
+ "hyperparameter": {
77
+ "pos_weight": 30.0,
78
+ "mbce": 1,
79
+ "pkd": 5,
80
+ "cont": 0.05
81
+ },
82
+ "lr_scheduler": {
83
+ "type": "WarmupPolyLR",
84
+ "args": {
85
+ "warmup_iters": 0.2
86
+ }
87
+ },
88
+ "evaluator": {
89
+ "type": "Evaluator",
90
+ "args": {}
91
+ },
92
+ "metrics": [
93
+ "Pixel_Accuracy",
94
+ "Pixel_Accuracy_Class",
95
+ "Mean_Intersection_over_Union"
96
+ ],
97
+ "trainer": {
98
+ "epochs": 100,
99
+ "save_dir": "saved_ade",
100
+ "save_period": 10,
101
+ "validation_period": 10,
102
+ "verbosity": 2,
103
+ "monitor": "off",
104
+ "early_stop": -1,
105
+ "reset_best_mnt": false,
106
+ "tensorboard": false
107
+ },
108
+ "test": false
109
+ }
ade_overlapped_50-50_STAR/overlapped_50-50_STAR.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bdcff89964d88c01f78d89e570c51ac8ce8761d7cd91ed009375ba06eed85ac
3
+ size 469579543