Skip to content

Commit a7f8bfb

Browse files
committed
add of fusion support for Semantic8 and NPM3D
1 parent 53e9138 commit a7f8bfb

File tree

8 files changed

+8682
-32
lines changed

8 files changed

+8682
-32
lines changed

examples/npm3d/README.md

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,4 +33,19 @@ python npm3d_seg.py --rootdir path_to_data_dir --savedir path_to_save_dir --test
3333
If the model was trained without using the lidar intensity:
3434
```
3535
python npm3d_seg.py --rootdir path_to_data_dir --savedir path_to_save_dir --test --nocolor
36+
```
37+
38+
**note**: the `test_step` parameter is set `0.8`. It is possible to change it. A smaller step of sliding window would produce better segmentation at a the cost of a longer computation time.
39+
40+
## Fusion model
41+
42+
Once models (RGB and without color information) have been trained, it is possible to train a fusion model.
43+
44+
### Training
45+
```
46+
python npm33d_seg_fusion.py --rootdir path_to_data_processed --savedir path_to_save_dirctory --model_rgb path_to_rgb_model_directory --model_noc path_to_no_color_model_directory
47+
```
48+
### Test
49+
```
50+
python npm3d_seg_fusion.py --rootdir path_to_data_processeed --savedir path_to_save_dirctory --model_rgb path_to_rgb_model_directory --model_noc path_to_no_color_model_directory --test --savepts
3651
```

examples/npm3d/npm3d_seg.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,10 +238,8 @@ def main():
238238

239239
# create the filelits (train / val) according to area
240240
print("Create filelist...", end="")
241-
filelist_train = []
242241
train_dir = os.path.join(args.rootdir, "train_pointclouds")
243242
filelist_train = [dataset for dataset in os.listdir(train_dir)]
244-
filename_test = []
245243
test_dir = os.path.join(args.rootdir, "test_pointclouds")
246244
filelist_test = [dataset for dataset in os.listdir(test_dir)]
247245
print(f"done, {len(filelist_train)} train files, {len(filelist_test)} test files")

0 commit comments

Comments
 (0)