Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cmd.md 2.32 KB
Newer Older
lishj6's avatar
init_0905  
lishj6 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66 
# Training cmd

## 1. FlashOcc
```shell script
bash tool/dist_train.sh projects/configs/flashocc/flashocc-r50-M0.py 4                             # 31.95
bash tool/dist_train.sh projects/configs/flashocc/flashocc-r50.py 4                                # 32.08
bash tool/dist_train.sh projects/configs/flashocc/flashocc-r50-4d-stereo.py 4                      # 37.84
bash tool/dist_train.sh projects/configs/flashocc/flashocc-stbase-4d-stereo-512x1408_4x4_1e-4.py 4 # 41.80
bash tool/dist_train.sh projects/configs/flashocc/flashocc-stbase-4d-stereo-512x1408_4x4_2e-4.py 4 # 43.52
```

## 2. Panoptic-FlashOcc
### for train
```shell script
conda activate FlashOcc
exp_name=panoptic-flashocc-r50-depth-tiny-pano
exp_name=panoptic-flashocc-r50-depth-pano
exp_name=panoptic-flashocc-r50-depth4d-pano
exp_name=panoptic-flashocc-r50-depth4d-longterm8f-pano
bash tools/dist_train.sh \
    projects/configs/panoptic-flashocc/${exp_name}.py \
    4
```

### for test
```shell script
conda activate FlashOcc
exp_name=panoptic-flashocc-r50-depth-tiny-pano
exp_name=panoptic-flashocc-r50-depth-pano
exp_name=panoptic-flashocc-r50-depth4d-pano
exp_name=panoptic-flashocc-r50-depth4d-longterm8f-pano
bash tools/dist_test.sh \
    projects/configs/panoptic-flashocc/${exp_name}.py \
    work_dirs/${exp_name}/epoch_24_ema.pth \
    4 \
    --eval ray-iou
```

### for vis
```shell script
exp_name=panoptic-flashocc-r50-depth-tiny-pano
exp_name=panoptic-flashocc-r50-depth-pano
exp_name=panoptic-flashocc-r50-depth4d-pano
exp_name=panoptic-flashocc-r50-depth4d-longterm8f-pano
python tools/vis_occ.py --config projects/configs/panoptic-flashocc/${exp_name}.py --weights work_dirs/${exp_name}/epoch_24_ema.pth --viz-dir vis/${exp_name} --draw-gt
```

### for test inference time
```shell script
conda activate FlashOcc
source activate FlashOcc
exp_name=panoptic-flashocc-r50-depth-tiny-pano
exp_name=panoptic-flashocc-r50-depth-pano
python tools/analysis_tools/benchmark.py \
    projects/configs/panoptic-flashocc/${exp_name}.py \
    work_dirs/${exp_name}/epoch_24_ema.pth \
    --w_pano --w_panoproc

exp_name=panoptic-flashocc-r50-depth4d-pano
exp_name=panoptic-flashocc-r50-depth4d-longterm8f-pano
python tools/analysis_tools/benchmark_sequential.py \
    projects/configs/panoptic-flashocc/${exp_name}.py \
    work_dirs/${exp_name}/epoch_24_ema.pth \
    --w_pano --w_panoproc

```