Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
zhangych02
7c7bccd00a
|
2 years ago | |
|---|---|---|
| .. | ||
| auction_match | 2 years ago | |
| chamfer_distance | 2 years ago | |
| checkpoints | 2 years ago | |
| data | 2 years ago | |
| evaluation_code | 2 years ago | |
| pointnet2 | 2 years ago | |
| tf_ops | 2 years ago | |
| utils | 2 years ago | |
| KNN_CUDA-0.2-py3-none-any.whl | 2 years ago | |
| README.md | 2 years ago | |
| data_loader.py | 2 years ago | |
| loss.py | 2 years ago | |
| model.py | 2 years ago | |
| test.py | 2 years ago | |
| train.py | 2 years ago | |
README.md
PU-GAN PyTorch version
Install KNN
pip install KNN_CUDA-0.2-py3-none-any.whl
Install Pointnet2 module
cd pointnet2
python setup.py install
dataset
The training data is put in folder ./data/train, and test data is in ./data/test.
training
python train.py
evaluation
please install CGAL firstly, and then run python test.py to generate prediction results in folder ./data/test/output, then copy the output into ./evaluation_code
cd evaluation
./evalution name.off name.xyz
No Description
Text Makefile C++ Python C other
MIT