diff --git a/README.md b/README.md index 7649eac..a0aecce 100644 --- a/README.md +++ b/README.md @@ -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