TensorBoard
karimknaebel commited on
Commit
284fd14
·
verified ·
1 Parent(s): 8277efe

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -45,3 +45,5 @@ scannet200/semseg-ptv3_dinov3-L/train.log filter=lfs diff=lfs merge=lfs -text
45
  scannet200/distill-ptv3_scannet200+structured3d_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
46
  scannet200/distill-ptv3_scannet200_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
47
  scannet200/semseg-ptv3_weight=distill-ptv3_scannet200+structured3d_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
 
 
 
45
  scannet200/distill-ptv3_scannet200+structured3d_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
46
  scannet200/distill-ptv3_scannet200_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
47
  scannet200/semseg-ptv3_weight=distill-ptv3_scannet200+structured3d_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
48
+ nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/test.log filter=lfs diff=lfs merge=lfs -text
49
+ nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/train.log filter=lfs diff=lfs merge=lfs -text
nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/config.py ADDED
@@ -0,0 +1,232 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ weight = 'exp/nuscenes/distill-ptv3dino-Llevels=0/model/model_last.pth'
2
+ resume = False
3
+ evaluate = True
4
+ test_only = False
5
+ seed = 32336762
6
+ save_path = 'exp/nuscenes/semseg-ptv3_weight=distill-L-levels=0'
7
+ wandb_project = 'semseg_nuscenes'
8
+ num_worker = 16
9
+ batch_size = 12
10
+ batch_size_val = None
11
+ batch_size_test = None
12
+ epoch = 50
13
+ eval_epoch = 50
14
+ clip_grad = None
15
+ sync_bn = False
16
+ enable_amp = True
17
+ empty_cache = False
18
+ empty_cache_per_epoch = False
19
+ find_unused_parameters = False
20
+ mix_prob = 0.8
21
+ param_dicts = [dict(keyword='block', lr=0.0002)]
22
+ hooks = [
23
+ dict(type='CheckpointLoader'),
24
+ dict(type='IterationTimer', warmup_iter=2),
25
+ dict(type='InformationWriter'),
26
+ dict(type='SemSegEvaluator'),
27
+ dict(type='CheckpointSaver', save_freq=None)
28
+ ]
29
+ train = dict(type='DefaultTrainer')
30
+ test = dict(type='SemSegTester', verbose=True)
31
+ model = dict(
32
+ type='DefaultSegmentorV2',
33
+ num_classes=16,
34
+ backbone_out_channels=64,
35
+ backbone=dict(
36
+ type='PT-v3m1',
37
+ in_channels=4,
38
+ order=['z', 'z-trans', 'hilbert', 'hilbert-trans'],
39
+ stride=(2, 2, 2, 2),
40
+ enc_depths=(2, 2, 2, 6, 2),
41
+ enc_channels=(32, 64, 128, 256, 512),
42
+ enc_num_head=(2, 4, 8, 16, 32),
43
+ enc_patch_size=(1024, 1024, 1024, 1024, 1024),
44
+ dec_depths=(2, 2, 2, 2),
45
+ dec_channels=(64, 64, 128, 256),
46
+ dec_num_head=(4, 4, 8, 16),
47
+ dec_patch_size=(1024, 1024, 1024, 1024),
48
+ mlp_ratio=4,
49
+ qkv_bias=True,
50
+ qk_scale=None,
51
+ attn_drop=0.0,
52
+ proj_drop=0.0,
53
+ drop_path=0.3,
54
+ shuffle_orders=True,
55
+ pre_norm=True,
56
+ enable_rpe=False,
57
+ enable_flash=True,
58
+ upcast_attention=False,
59
+ upcast_softmax=False,
60
+ cls_mode=False,
61
+ pdnorm_bn=False,
62
+ pdnorm_ln=False,
63
+ pdnorm_decouple=True,
64
+ pdnorm_adaptive=False,
65
+ pdnorm_affine=True,
66
+ pdnorm_conditions=('nuScenes', 'SemanticKITTI', 'Waymo')),
67
+ criteria=[
68
+ dict(type='CrossEntropyLoss', loss_weight=1.0, ignore_index=-1),
69
+ dict(
70
+ type='LovaszLoss',
71
+ mode='multiclass',
72
+ loss_weight=1.0,
73
+ ignore_index=-1)
74
+ ])
75
+ optimizer = dict(type='AdamW', lr=0.002, weight_decay=0.005)
76
+ scheduler = dict(
77
+ type='OneCycleLR',
78
+ max_lr=[0.002, 0.0002],
79
+ pct_start=0.04,
80
+ anneal_strategy='cos',
81
+ div_factor=10.0,
82
+ final_div_factor=100.0)
83
+ dataset_type = 'NuScenesDataset'
84
+ data_root = 'data/nuscenes'
85
+ ignore_index = -1
86
+ names = [
87
+ 'barrier', 'bicycle', 'bus', 'car', 'construction_vehicle', 'motorcycle',
88
+ 'pedestrian', 'traffic_cone', 'trailer', 'truck', 'driveable_surface',
89
+ 'other_flat', 'sidewalk', 'terrain', 'manmade', 'vegetation'
90
+ ]
91
+ data = dict(
92
+ num_classes=16,
93
+ ignore_index=-1,
94
+ names=[
95
+ 'barrier', 'bicycle', 'bus', 'car', 'construction_vehicle',
96
+ 'motorcycle', 'pedestrian', 'traffic_cone', 'trailer', 'truck',
97
+ 'driveable_surface', 'other_flat', 'sidewalk', 'terrain', 'manmade',
98
+ 'vegetation'
99
+ ],
100
+ train=dict(
101
+ type='NuScenesDataset',
102
+ split='train',
103
+ data_root='data/nuscenes',
104
+ transform=[
105
+ dict(
106
+ type='RandomRotate',
107
+ angle=[-1, 1],
108
+ axis='z',
109
+ center=[0, 0, 0],
110
+ p=0.5),
111
+ dict(type='RandomScale', scale=[0.9, 1.1]),
112
+ dict(type='RandomFlip', p=0.5),
113
+ dict(type='RandomJitter', sigma=0.005, clip=0.02),
114
+ dict(
115
+ type='GridSample',
116
+ grid_size=0.05,
117
+ hash_type='fnv',
118
+ mode='train',
119
+ keys=('coord', 'strength', 'segment'),
120
+ return_grid_coord=True),
121
+ dict(type='ToTensor'),
122
+ dict(
123
+ type='Collect',
124
+ keys=('coord', 'grid_coord', 'segment'),
125
+ feat_keys=('coord', 'strength'))
126
+ ],
127
+ test_mode=False,
128
+ ignore_index=-1,
129
+ loop=1),
130
+ val=dict(
131
+ type='NuScenesDataset',
132
+ split='val',
133
+ data_root='data/nuscenes',
134
+ transform=[
135
+ dict(
136
+ type='GridSample',
137
+ grid_size=0.05,
138
+ hash_type='fnv',
139
+ mode='train',
140
+ keys=('coord', 'strength', 'segment'),
141
+ return_grid_coord=True),
142
+ dict(type='ToTensor'),
143
+ dict(
144
+ type='Collect',
145
+ keys=('coord', 'grid_coord', 'segment'),
146
+ feat_keys=('coord', 'strength'))
147
+ ],
148
+ test_mode=False,
149
+ ignore_index=-1),
150
+ test=dict(
151
+ type='NuScenesDataset',
152
+ split='val',
153
+ data_root='data/nuscenes',
154
+ transform=[
155
+ dict(type='Copy', keys_dict=dict(segment='origin_segment')),
156
+ dict(
157
+ type='GridSample',
158
+ grid_size=0.025,
159
+ hash_type='fnv',
160
+ mode='train',
161
+ keys=('coord', 'strength', 'segment'),
162
+ return_inverse=True)
163
+ ],
164
+ test_mode=True,
165
+ test_cfg=dict(
166
+ voxelize=dict(
167
+ type='GridSample',
168
+ grid_size=0.05,
169
+ hash_type='fnv',
170
+ mode='test',
171
+ return_grid_coord=True,
172
+ keys=('coord', 'strength')),
173
+ crop=None,
174
+ post_transform=[
175
+ dict(type='ToTensor'),
176
+ dict(
177
+ type='Collect',
178
+ keys=('coord', 'grid_coord', 'index'),
179
+ feat_keys=('coord', 'strength'))
180
+ ],
181
+ aug_transform=[[{
182
+ 'type': 'RandomScale',
183
+ 'scale': [0.9, 0.9]
184
+ }], [{
185
+ 'type': 'RandomScale',
186
+ 'scale': [0.95, 0.95]
187
+ }], [{
188
+ 'type': 'RandomScale',
189
+ 'scale': [1, 1]
190
+ }], [{
191
+ 'type': 'RandomScale',
192
+ 'scale': [1.05, 1.05]
193
+ }], [{
194
+ 'type': 'RandomScale',
195
+ 'scale': [1.1, 1.1]
196
+ }],
197
+ [{
198
+ 'type': 'RandomScale',
199
+ 'scale': [0.9, 0.9]
200
+ }, {
201
+ 'type': 'RandomFlip',
202
+ 'p': 1
203
+ }],
204
+ [{
205
+ 'type': 'RandomScale',
206
+ 'scale': [0.95, 0.95]
207
+ }, {
208
+ 'type': 'RandomFlip',
209
+ 'p': 1
210
+ }],
211
+ [{
212
+ 'type': 'RandomScale',
213
+ 'scale': [1, 1]
214
+ }, {
215
+ 'type': 'RandomFlip',
216
+ 'p': 1
217
+ }],
218
+ [{
219
+ 'type': 'RandomScale',
220
+ 'scale': [1.05, 1.05]
221
+ }, {
222
+ 'type': 'RandomFlip',
223
+ 'p': 1
224
+ }],
225
+ [{
226
+ 'type': 'RandomScale',
227
+ 'scale': [1.1, 1.1]
228
+ }, {
229
+ 'type': 'RandomFlip',
230
+ 'p': 1
231
+ }]]),
232
+ ignore_index=-1))
nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/events.out.tfevents.1730063991.n23g0005.hpc.itc.rwth-aachen.de ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5078840bbed63f6efb78a9a802454de4d6ee64defd3a2924b896c63f33d7091
3
+ size 11464320
nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/model/model_best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93dbb9d3fda247adf56e1c1ebd419d6ada4ebe4dc141110c606926d3ad67b30d
3
+ size 554518952
nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/model/model_last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d8a8955dc0e69f9173c3a398d06330d7e8401f8fa39c7b167187a60d1509606
3
+ size 554518952
nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/test.log ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03e69ba744eab06993383861d1c428b91d7f01a7f19017655afdaaee853861c7
3
+ size 33185885
nuscenes/semseg-ptv3_weight=distill-ptv3_nuscenes_dino-L/train.log ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7cb49c17d46d3a631a3259d283bc2b9dad8ff7911be504512bb1c2b857db6f8
3
+ size 32589918