Update README.md

This commit is contained in:
Xu Ma 2021-10-04 12:15:43 -04:00 committed by GitHub
parent 424fa7ca7b
commit 8e5c90481a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -16,7 +16,7 @@ pip install pointnet2_ops_lib/.
```
## Classification ModelNet40
The dataset will be automatically downloaded, directly run following command to train
The dataset will be automatically downloaded, run following command to train
```bash
# train pointMLP
python main.py --model pointMLP
@ -24,6 +24,13 @@ python main.py --model pointMLP
python main.py --model pointMLPElite
# please add other paramemters as you wish.
```
By default, it will create a fold named "checkpoints/{modelName}-{msg}-{randomseed}", which includes args.txt, best_checkpoint.pth, last_checkpoint.pth, log.txt, out.txt.
To conduct voting experiments, run
```bash
# please modify the msg accrodingly
python voting.py --model pointMLP --msg demo
```
## Classification ScanObjectNN