Skip to content

Commit bb97aa7

Browse files
author
liubigli
committed
conf files updated to hydra 1.1
1 parent 6bf40ba commit bb97aa7

File tree

92 files changed

+4980
-6457
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

92 files changed

+4980
-6457
lines changed

conf/config.yaml

+9-14
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,16 @@
1-
defaults: # for loading the default config
2-
- task: ??? # Task performed (segmentation, classification etc...)
3-
optional: True
4-
- model_type: ??? # Type of model to use, e.g. pointnet2, rsconv etc...
5-
optional: True
6-
- dataset: ???
7-
optional: True
1+
defaults: # for loading the default.yaml config
2+
- task: ???
83

94
- visualization: default
105
- lr_scheduler: exponential
116
- training: default
12-
13-
- debugging: default.yaml
14-
- models: ${defaults.0.task}/${defaults.1.model_type}
15-
- data: ${defaults.0.task}/${defaults.2.dataset}
7+
#
8+
- debugging: default
9+
- models: ???
10+
- data: ???
1611
- sota # Contains current SOTA results on different datasets (extracted from papers !).
17-
- hydra/job_logging: custom
18-
- hydra/output: custom # add the support for user-defined experiment folder (where to save the experiment files)
12+
# - hydra/job_logging: custom
13+
# - hydra/output: custom # add the support for user-defined experiment folder (where to save the experiment files)
1914

2015
job_name: benchmark # prefix name for saving the experiment file.
2116
model_name: ??? # Name of the specific model to load
@@ -27,4 +22,4 @@ eval_frequency: 1
2722
tracker_options: # Extra options for the tracker
2823
full_res: False
2924
make_submission: False
30-
track_boxes: False
25+
track_boxes: False
+18-16
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,18 @@
1-
data:
2-
task: classification
3-
class: modelnet.ModelNetDataset
4-
name: modelnet
5-
dataroot: data
6-
number: 10
7-
pre_transforms:
8-
- transform: NormalizeScale
9-
- transform: GridSampling3D
10-
lparams: [0.02]
11-
train_transforms:
12-
- transform: FixedPoints
13-
lparams: [2048]
14-
test_transforms:
15-
- transform: FixedPoints
16-
lparams: [2048]
1+
# @package data
2+
defaults:
3+
- classification/default
4+
5+
class: modelnet.ModelNetDataset
6+
name: modelnet
7+
dataroot: data
8+
number: 10
9+
pre_transforms:
10+
- transform: NormalizeScale
11+
- transform: GridSampling3D
12+
lparams: [0.02]
13+
train_transforms:
14+
- transform: FixedPoints
15+
lparams: [2048]
16+
test_transforms:
17+
- transform: FixedPoints
18+
lparams: [2048]
+33-31
Original file line numberDiff line numberDiff line change
@@ -1,33 +1,35 @@
1-
data:
2-
class: scannet.ScannetDataset
3-
dataset_name: "scannet-sparse"
4-
task: object_detection
5-
dataroot: data
6-
version: "v2"
7-
use_instance_labels: True
8-
use_instance_bboxes: True
9-
donotcare_class_ids: []
10-
process_workers: 8
1+
# @package data
2+
defaults:
3+
- object_detection/default
114

12-
train_transform:
13-
- transform: FixedPoints
14-
lparams: [50000]
15-
- transform: RandomNoise
16-
params:
17-
sigma: 0.01
18-
clip: 0.05
19-
- transform: RandomScaleAnisotropic
20-
params:
21-
scales: [0.9, 1.1]
22-
- transform: RandomSymmetry
23-
params:
24-
axis: [True, False, False]
25-
- transform: Random3AxisRotation
26-
params:
27-
rot_x: 2
28-
rot_y: 2
29-
rot_z: 180
5+
class: scannet.ScannetDataset
6+
dataset_name: "scannet-sparse"
7+
dataroot: data
8+
version: "v2"
9+
use_instance_labels: True
10+
use_instance_bboxes: True
11+
donotcare_class_ids: []
12+
process_workers: 8
3013

31-
val_transform:
32-
- transform: FixedPoints
33-
lparams: [50000]
14+
train_transform:
15+
- transform: FixedPoints
16+
lparams: [50000]
17+
- transform: RandomNoise
18+
params:
19+
sigma: 0.01
20+
clip: 0.05
21+
- transform: RandomScaleAnisotropic
22+
params:
23+
scales: [0.9, 1.1]
24+
- transform: RandomSymmetry
25+
params:
26+
axis: [True, False, False]
27+
- transform: Random3AxisRotation
28+
params:
29+
rot_x: 2
30+
rot_y: 2
31+
rot_z: 180
32+
33+
val_transform:
34+
- transform: FixedPoints
35+
lparams: [50000]
+73-71
Original file line numberDiff line numberDiff line change
@@ -1,75 +1,77 @@
1-
data:
2-
class: scannet.ScannetDataset
3-
dataset_name: "scannet-sparse"
4-
task: object_detection
5-
dataroot: data
6-
version: "v2"
7-
use_instance_labels: True
8-
use_instance_bboxes: True
9-
donotcare_class_ids: []
10-
process_workers: 8
1+
# @package data
2+
defaults:
3+
- object_detection/default
114

12-
grid_size: 0.05
13-
mode: last
5+
class: scannet.ScannetDataset
6+
dataset_name: "scannet-sparse"
7+
dataroot: data
8+
version: "v2"
9+
use_instance_labels: True
10+
use_instance_bboxes: True
11+
donotcare_class_ids: []
12+
process_workers: 8
1413

15-
pre_transform:
16-
- transform: GridSampling3D
17-
lparams: [0.02]
14+
grid_size: 0.05
15+
mode: last
1816

19-
train_transform:
20-
- transform: XYZFeature
21-
params:
22-
add_x: True
23-
add_y: True
24-
add_z: True
25-
- transform: RandomNoise
26-
params:
27-
sigma: 0.01
28-
clip: 0.05
29-
- transform: RandomScaleAnisotropic
30-
params:
31-
scales: [0.9, 1.1]
32-
- transform: RandomSymmetry
33-
params:
34-
axis: [True, False, False]
35-
- transform: Random3AxisRotation
36-
params:
37-
rot_x: 2
38-
rot_y: 2
39-
rot_z: 180
40-
- transform: GridSampling3D
41-
params:
42-
size: ${data.grid_size}
43-
quantize_coords: True
44-
mode: ${data.mode}
45-
- transform: FixedPoints
46-
lparams: [50000]
47-
params:
48-
replace: False
49-
- transform: ShiftVoxels
50-
- transform: AddFeatsByKeys
51-
params:
52-
list_add_to_x: [True, True, True]
53-
feat_names: ["pos_x", "pos_y", "pos_z"]
54-
delete_feats: [True, True, True]
17+
pre_transform:
18+
- transform: GridSampling3D
19+
lparams: [0.02]
5520

56-
val_transform:
57-
- transform: XYZFeature
58-
params:
59-
add_x: True
60-
add_y: True
61-
add_z: True
62-
- transform: GridSampling3D
63-
params:
64-
size: ${data.grid_size}
65-
quantize_coords: True
66-
mode: ${data.mode}
67-
- transform: FixedPoints
68-
lparams: [50000]
69-
params:
70-
replace: False
71-
- transform: AddFeatsByKeys
72-
params:
73-
list_add_to_x: [True, True, True]
74-
feat_names: ["pos_x", "pos_y", "pos_z"]
75-
delete_feats: [True, True, True]
21+
train_transform:
22+
- transform: XYZFeature
23+
params:
24+
add_x: True
25+
add_y: True
26+
add_z: True
27+
- transform: RandomNoise
28+
params:
29+
sigma: 0.01
30+
clip: 0.05
31+
- transform: RandomScaleAnisotropic
32+
params:
33+
scales: [0.9, 1.1]
34+
- transform: RandomSymmetry
35+
params:
36+
axis: [True, False, False]
37+
- transform: Random3AxisRotation
38+
params:
39+
rot_x: 2
40+
rot_y: 2
41+
rot_z: 180
42+
- transform: GridSampling3D
43+
params:
44+
size: ${data.grid_size}
45+
quantize_coords: True
46+
mode: ${data.mode}
47+
- transform: FixedPoints
48+
lparams: [50000]
49+
params:
50+
replace: False
51+
- transform: ShiftVoxels
52+
- transform: AddFeatsByKeys
53+
params:
54+
list_add_to_x: [True, True, True]
55+
feat_names: ["pos_x", "pos_y", "pos_z"]
56+
delete_feats: [True, True, True]
57+
58+
val_transform:
59+
- transform: XYZFeature
60+
params:
61+
add_x: True
62+
add_y: True
63+
add_z: True
64+
- transform: GridSampling3D
65+
params:
66+
size: ${data.grid_size}
67+
quantize_coords: True
68+
mode: ${data.mode}
69+
- transform: FixedPoints
70+
lparams: [50000]
71+
params:
72+
replace: False
73+
- transform: AddFeatsByKeys
74+
params:
75+
list_add_to_x: [True, True, True]
76+
feat_names: ["pos_x", "pos_y", "pos_z"]
77+
delete_feats: [True, True, True]

0 commit comments

Comments
 (0)