Update README.md

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

View file

@ -16,6 +16,15 @@ pip install pointnet2_ops_lib/.
```
## Classification ModelNet40
The dataset will be automatically downloaded, directly run following command to train
```bash
# train pointMLP
python main.py --model pointMLP
# train pointMLP-elite
python main.py --model pointMLPElite
# please add other paramemters as you wish.
```
## Classification ScanObjectNN