├── LICENSE ├── README.md ├── args ├── pu1k_args.py ├── pugan_args.py └── utils.py ├── data ├── PU-GAN │ └── train │ │ └── .gitkeep └── PU1K │ └── .gitkeep ├── dataset ├── dataset.py └── utils.py ├── evaluation_code ├── CMakeLists.txt ├── Icosahedron.off ├── Icosahedron.xyz ├── __init__.py ├── compile.sh ├── evaluation.cpp ├── readme.txt ├── utils │ ├── eulerangles.py │ ├── logger.py │ ├── misc.py │ ├── model_utils.py │ ├── pc_util.py │ └── tf_util.py └── write_eval_script.py ├── models ├── Chamfer3D │ ├── chamfer3D.cu │ ├── chamfer_cuda.cpp │ ├── dist_chamfer_3D.py │ └── setup.py ├── FeatureExtractor.py ├── P2PNet.py ├── P2PRegressor.py ├── pointops │ ├── __init__.py │ ├── functions │ │ ├── __init__.py │ │ └── pointops.py │ ├── setup.py │ └── src │ │ ├── __init__.py │ │ ├── ballquery │ │ ├── ballquery_cuda.cpp │ │ ├── ballquery_cuda_kernel.cu │ │ └── ballquery_cuda_kernel.h │ │ ├── cuda_utils.h │ │ ├── featuredistribute │ │ ├── featuredistribute_cuda.cpp │ │ ├── featuredistribute_cuda_kernel.cu │ │ └── featuredistribute_cuda_kernel.h │ │ ├── grouping │ │ ├── grouping_cuda.cpp │ │ ├── grouping_cuda_kernel.cu │ │ └── grouping_cuda_kernel.h │ │ ├── grouping_int │ │ ├── grouping_int_cuda.cpp │ │ ├── grouping_int_cuda_kernel.cu │ │ └── grouping_int_cuda_kernel.h │ │ ├── interpolation │ │ ├── interpolation_cuda.cpp │ │ ├── interpolation_cuda_kernel.cu │ │ └── interpolation_cuda_kernel.h │ │ ├── knnquery │ │ ├── __init__.py │ │ ├── knnquery_cuda.cpp │ │ ├── knnquery_cuda_kernel.cu │ │ └── knnquery_cuda_kernel.h │ │ ├── knnquery_heap │ │ ├── __init__.py │ │ ├── knnquery_heap_cuda.cpp │ │ ├── knnquery_heap_cuda_kernel.cu │ │ └── knnquery_heap_cuda_kernel.h │ │ ├── labelstat │ │ ├── labelstat_cuda.cpp │ │ ├── labelstat_cuda_kernel.cu │ │ └── labelstat_cuda_kernel.h │ │ ├── pointops_api.cpp │ │ └── sampling │ │ ├── sampling_cuda.cpp │ │ ├── sampling_cuda_kernel.cu │ │ └── sampling_cuda_kernel.h └── utils.py ├── p2pnet.png ├── prepare_pugan.py ├── pretrained_model ├── pu1k │ └── ckpt │ │ └── ckpt-epoch-60.pth └── pugan │ └── ckpt │ └── ckpt-epoch-60.pth ├── test.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/README.md -------------------------------------------------------------------------------- /args/pu1k_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/args/pu1k_args.py -------------------------------------------------------------------------------- /args/pugan_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/args/pugan_args.py -------------------------------------------------------------------------------- /args/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/args/utils.py -------------------------------------------------------------------------------- /data/PU-GAN/train/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/PU1K/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/dataset/dataset.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /evaluation_code/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/CMakeLists.txt -------------------------------------------------------------------------------- /evaluation_code/Icosahedron.off: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/Icosahedron.off -------------------------------------------------------------------------------- /evaluation_code/Icosahedron.xyz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/Icosahedron.xyz -------------------------------------------------------------------------------- /evaluation_code/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/__init__.py -------------------------------------------------------------------------------- /evaluation_code/compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/compile.sh -------------------------------------------------------------------------------- /evaluation_code/evaluation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/evaluation.cpp -------------------------------------------------------------------------------- /evaluation_code/readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/readme.txt -------------------------------------------------------------------------------- /evaluation_code/utils/eulerangles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/utils/eulerangles.py -------------------------------------------------------------------------------- /evaluation_code/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/utils/logger.py -------------------------------------------------------------------------------- /evaluation_code/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/utils/misc.py -------------------------------------------------------------------------------- /evaluation_code/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/utils/model_utils.py -------------------------------------------------------------------------------- /evaluation_code/utils/pc_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/utils/pc_util.py -------------------------------------------------------------------------------- /evaluation_code/utils/tf_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/utils/tf_util.py -------------------------------------------------------------------------------- /evaluation_code/write_eval_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/evaluation_code/write_eval_script.py -------------------------------------------------------------------------------- /models/Chamfer3D/chamfer3D.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/Chamfer3D/chamfer3D.cu -------------------------------------------------------------------------------- /models/Chamfer3D/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/Chamfer3D/chamfer_cuda.cpp -------------------------------------------------------------------------------- /models/Chamfer3D/dist_chamfer_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/Chamfer3D/dist_chamfer_3D.py -------------------------------------------------------------------------------- /models/Chamfer3D/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/Chamfer3D/setup.py -------------------------------------------------------------------------------- /models/FeatureExtractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/FeatureExtractor.py -------------------------------------------------------------------------------- /models/P2PNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/P2PNet.py -------------------------------------------------------------------------------- /models/P2PRegressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/P2PRegressor.py -------------------------------------------------------------------------------- /models/pointops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/functions/pointops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/functions/pointops.py -------------------------------------------------------------------------------- /models/pointops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/setup.py -------------------------------------------------------------------------------- /models/pointops/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/src/ballquery/ballquery_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/ballquery/ballquery_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/ballquery/ballquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/ballquery/ballquery_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/ballquery/ballquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/ballquery/ballquery_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/cuda_utils.h -------------------------------------------------------------------------------- /models/pointops/src/featuredistribute/featuredistribute_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/featuredistribute/featuredistribute_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/grouping/grouping_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/grouping/grouping_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/grouping/grouping_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/grouping/grouping_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/grouping/grouping_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/grouping/grouping_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/grouping_int/grouping_int_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/grouping_int/grouping_int_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/grouping_int/grouping_int_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/grouping_int/grouping_int_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/grouping_int/grouping_int_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/grouping_int/grouping_int_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/interpolation/interpolation_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/interpolation/interpolation_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/interpolation/interpolation_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/interpolation/interpolation_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/interpolation/interpolation_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/interpolation/interpolation_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/knnquery/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/src/knnquery/knnquery_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/knnquery/knnquery_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/knnquery/knnquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/knnquery/knnquery_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/knnquery/knnquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/knnquery/knnquery_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/knnquery_heap/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/src/knnquery_heap/knnquery_heap_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/knnquery_heap/knnquery_heap_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/labelstat/labelstat_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/labelstat/labelstat_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/labelstat/labelstat_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/labelstat/labelstat_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/labelstat/labelstat_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/labelstat/labelstat_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/pointops_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/pointops_api.cpp -------------------------------------------------------------------------------- /models/pointops/src/sampling/sampling_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/sampling/sampling_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/sampling/sampling_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/sampling/sampling_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/sampling/sampling_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/pointops/src/sampling/sampling_cuda_kernel.h -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/models/utils.py -------------------------------------------------------------------------------- /p2pnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/p2pnet.png -------------------------------------------------------------------------------- /prepare_pugan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/prepare_pugan.py -------------------------------------------------------------------------------- /pretrained_model/pu1k/ckpt/ckpt-epoch-60.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/pretrained_model/pu1k/ckpt/ckpt-epoch-60.pth -------------------------------------------------------------------------------- /pretrained_model/pugan/ckpt/ckpt-epoch-60.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/pretrained_model/pugan/ckpt/ckpt-epoch-60.pth -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yunhe20/Grad-PU/HEAD/train.py --------------------------------------------------------------------------------