├── README.md ├── assets └── gpat_teaser.jpg ├── dataset ├── gen_st_inds.npy ├── non_exact_parts │ ├── rectangles.npy │ └── spheres.npy ├── preprocess.py ├── rand_quats.npy ├── rand_type.npy ├── test_st_inds.npy └── upsample.py ├── environment.yml ├── learning ├── assembler.py ├── dataset.py ├── gpat │ ├── gpat.py │ └── run.py ├── learning_utils.py ├── modules.py ├── opt │ ├── optimizer.py │ └── run.py └── pointnet.py └── utils ├── __init__.py ├── chamfer ├── README.md ├── __init__.py ├── chamfer.py ├── cuda │ ├── chamfer.cpp │ └── chamfer_kernel.cu └── setup.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 /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/README.md -------------------------------------------------------------------------------- /assets/gpat_teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/assets/gpat_teaser.jpg -------------------------------------------------------------------------------- /dataset/gen_st_inds.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/gen_st_inds.npy -------------------------------------------------------------------------------- /dataset/non_exact_parts/rectangles.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/non_exact_parts/rectangles.npy -------------------------------------------------------------------------------- /dataset/non_exact_parts/spheres.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/non_exact_parts/spheres.npy -------------------------------------------------------------------------------- /dataset/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/preprocess.py -------------------------------------------------------------------------------- /dataset/rand_quats.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/rand_quats.npy -------------------------------------------------------------------------------- /dataset/rand_type.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/rand_type.npy -------------------------------------------------------------------------------- /dataset/test_st_inds.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/test_st_inds.npy -------------------------------------------------------------------------------- /dataset/upsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/dataset/upsample.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/environment.yml -------------------------------------------------------------------------------- /learning/assembler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/assembler.py -------------------------------------------------------------------------------- /learning/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/dataset.py -------------------------------------------------------------------------------- /learning/gpat/gpat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/gpat/gpat.py -------------------------------------------------------------------------------- /learning/gpat/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/gpat/run.py -------------------------------------------------------------------------------- /learning/learning_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/learning_utils.py -------------------------------------------------------------------------------- /learning/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/modules.py -------------------------------------------------------------------------------- /learning/opt/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/opt/optimizer.py -------------------------------------------------------------------------------- /learning/opt/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/opt/run.py -------------------------------------------------------------------------------- /learning/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/learning/pointnet.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/chamfer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/chamfer/README.md -------------------------------------------------------------------------------- /utils/chamfer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/chamfer/chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/chamfer/chamfer.py -------------------------------------------------------------------------------- /utils/chamfer/cuda/chamfer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/chamfer/cuda/chamfer.cpp -------------------------------------------------------------------------------- /utils/chamfer/cuda/chamfer_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/chamfer/cuda/chamfer_kernel.cu -------------------------------------------------------------------------------- /utils/chamfer/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/chamfer/setup.py -------------------------------------------------------------------------------- /utils/pointops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/pointops/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/pointops/functions/pointops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/functions/pointops.py -------------------------------------------------------------------------------- /utils/pointops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/setup.py -------------------------------------------------------------------------------- /utils/pointops/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/pointops/src/ballquery/ballquery_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/ballquery/ballquery_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/ballquery/ballquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/ballquery/ballquery_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/ballquery/ballquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/ballquery/ballquery_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/cuda_utils.h -------------------------------------------------------------------------------- /utils/pointops/src/featuredistribute/featuredistribute_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/featuredistribute/featuredistribute_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/featuredistribute/featuredistribute_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/featuredistribute/featuredistribute_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/featuredistribute/featuredistribute_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/featuredistribute/featuredistribute_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/grouping/grouping_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/grouping/grouping_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/grouping/grouping_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/grouping/grouping_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/grouping/grouping_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/grouping/grouping_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/grouping_int/grouping_int_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/grouping_int/grouping_int_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/grouping_int/grouping_int_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/grouping_int/grouping_int_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/grouping_int/grouping_int_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/grouping_int/grouping_int_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/interpolation/interpolation_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/interpolation/interpolation_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/interpolation/interpolation_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/interpolation/interpolation_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/interpolation/interpolation_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/interpolation/interpolation_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/knnquery/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/pointops/src/knnquery/knnquery_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/knnquery/knnquery_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/knnquery/knnquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/knnquery/knnquery_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/knnquery/knnquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/knnquery/knnquery_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/knnquery_heap/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/pointops/src/knnquery_heap/knnquery_heap_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/knnquery_heap/knnquery_heap_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/labelstat/labelstat_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/labelstat/labelstat_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/labelstat/labelstat_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/labelstat/labelstat_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/labelstat/labelstat_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/labelstat/labelstat_cuda_kernel.h -------------------------------------------------------------------------------- /utils/pointops/src/pointops_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/pointops_api.cpp -------------------------------------------------------------------------------- /utils/pointops/src/sampling/sampling_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/sampling/sampling_cuda.cpp -------------------------------------------------------------------------------- /utils/pointops/src/sampling/sampling_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/sampling/sampling_cuda_kernel.cu -------------------------------------------------------------------------------- /utils/pointops/src/sampling/sampling_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/pointops/src/sampling/sampling_cuda_kernel.h -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/real-stanford/gpat/HEAD/utils/utils.py --------------------------------------------------------------------------------