Here is the VRNet code.
The code is tested with Pytorch 1.6.0 with CUDA 10.2.89. Prerequisites include scipy, h5py, tqdm, etc. Your can install them by yourself.
The ModelNet40 dataset can be download from:
https://github.com/WangYueFt/dcp
Start training with the command:
python main.py
Start testing with the command:
python main.py --eval True --mdoel_path YOUR_CHECKPOINT_DIRECTORY
The code is insipred by DCP, PRNet, RPMNet, etc.