Skip to content

Commit b644324

Browse files
authored
Update README.md
1 parent 8fd2159 commit b644324

File tree

1 file changed

+2
-20
lines changed

1 file changed

+2
-20
lines changed

README.md

+2-20
Original file line numberDiff line numberDiff line change
@@ -88,30 +88,12 @@ ln -s /PATH/TO/MyData/dataset data
8888
### Train
8989
The following command is fit with a 12GB GPU.
9090
```bash
91-
python train.py --batch_size=2 --epoch=100 --inner_iter=5 --pcd_sample=20000 --name=cam2_muliter
91+
python train.py --batch_size=8 --epoch=100 --inner_iter=1 --pcd_sample=4096 --name=cam2_oneiter
9292
```
93-
<details>
94-
<summary>Recommended training strategy</summary>
95-
96-
A more successful way is to train the `one-iter` model first and then train the `multi-iter` one with the pretrained weigths of `one-iter` model.
97-
```bash
98-
python train.py --inner_iter=1 --name=cam2_oneiter --skip_frame=30 --pcd_sample=20000
99-
python train.py --inner_iter=5 --pretrained=./checkpoint/cam2_oneiter_best.pth --name=cam2_muliter --skip_frame=30 --pcd_sample=20000
100-
```
101-
102-
Similar to [mmsegmentation](https://github.com/open-mmlab/mmsegmentation/blob/master/docs/en/model_zoo.md), We use ResNetV1c instead of ResNetV1b.
103-
104-
However, we don't use pretrained weights because it isn't effective enough. If you wan't to use pretrained resnet18, please see [https://download.openmmlab.com/pretrain/third_party/resnet18_v1c-b5776b93.pth](https://download.openmmlab.com/pretrain/third_party/resnet18_v1c-b5776b93.pth).
105-
106-
Relevant training logs can be found in [log](./log) dir.
107-
108-
Try to set `skip_frame=5` or smaller to enlarge datasets if you have achieved similar results to our logs with `skip_frame=30`.
109-
110-
</details>
11193

11294
### Test
11395
```bash
114-
python test.py --batch_size=1 --inner_iter=5 --pretrained=./checkpoint/cam2_muliter_best.pth --skip_frame=1 --pcd_sample=-1
96+
python test.py --inner_iter=1 --pretrained=./checkpoint/cam2_oneiter_best.pth --skip_frame=1 --pcd_sample=-1
11597
```
11698
`pcd_sample=-1` means totally sample (but disorder) the raw pont cloud. However, you need to keep `batch_size=1` to avoid batch collect_fn error.
11799

0 commit comments

Comments
 (0)