├── .gitignore ├── LICENSE ├── README.md ├── cfgs ├── upsampling │ ├── __init__.py │ ├── pu1k_args.py │ ├── pugan_args.py │ └── pugan_paper_args.py └── utils.py ├── dataset ├── dataset.py └── utils.py ├── models ├── Chamfer3D │ ├── chamfer3D.cu │ ├── chamfer_cuda.cpp │ ├── dist_chamfer_3D.py │ └── setup.py ├── README.md ├── decoder │ ├── __init__.py │ ├── kernel.py │ └── kernel_o.py ├── encoder │ ├── __init__.py │ └── pointtrans.py ├── kernel_points │ ├── dispositions │ │ ├── k_002_center.ply │ │ ├── k_003_center.ply │ │ ├── k_004_center.ply │ │ ├── k_005_center.ply │ │ ├── k_006_center.ply │ │ ├── k_007_center.ply │ │ ├── k_008_center.ply │ │ ├── k_009_center.ply │ │ ├── k_010_center.ply │ │ ├── k_011_center.ply │ │ ├── k_012_center.ply │ │ ├── k_013_center.ply │ │ ├── k_014_center.ply │ │ ├── k_015_center.ply │ │ ├── k_016_center.ply │ │ └── k_017_center.ply │ ├── kernel_utils.py │ └── plyutils.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 ├── repkpu.py └── utils.py ├── surf_recon.py ├── test.py ├── train.py └── vis.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/upsampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/cfgs/upsampling/__init__.py -------------------------------------------------------------------------------- /cfgs/upsampling/pu1k_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/cfgs/upsampling/pu1k_args.py -------------------------------------------------------------------------------- /cfgs/upsampling/pugan_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/cfgs/upsampling/pugan_args.py -------------------------------------------------------------------------------- /cfgs/upsampling/pugan_paper_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/cfgs/upsampling/pugan_paper_args.py -------------------------------------------------------------------------------- /cfgs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/cfgs/utils.py -------------------------------------------------------------------------------- /dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/dataset/dataset.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /models/Chamfer3D/chamfer3D.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/Chamfer3D/chamfer3D.cu -------------------------------------------------------------------------------- /models/Chamfer3D/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/Chamfer3D/chamfer_cuda.cpp -------------------------------------------------------------------------------- /models/Chamfer3D/dist_chamfer_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/Chamfer3D/dist_chamfer_3D.py -------------------------------------------------------------------------------- /models/Chamfer3D/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/Chamfer3D/setup.py -------------------------------------------------------------------------------- /models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/README.md -------------------------------------------------------------------------------- /models/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/decoder/__init__.py -------------------------------------------------------------------------------- /models/decoder/kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/decoder/kernel.py -------------------------------------------------------------------------------- /models/decoder/kernel_o.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/decoder/kernel_o.py -------------------------------------------------------------------------------- /models/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/encoder/__init__.py -------------------------------------------------------------------------------- /models/encoder/pointtrans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/encoder/pointtrans.py -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_002_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_002_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_003_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_003_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_004_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_004_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_005_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_005_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_006_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_006_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_007_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_007_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_008_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_008_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_009_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_009_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_010_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_010_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_011_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_011_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_012_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_012_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_013_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_013_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_014_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_014_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_015_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_015_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_016_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_016_center.ply -------------------------------------------------------------------------------- /models/kernel_points/dispositions/k_017_center.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/dispositions/k_017_center.ply -------------------------------------------------------------------------------- /models/kernel_points/kernel_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/kernel_utils.py -------------------------------------------------------------------------------- /models/kernel_points/plyutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/kernel_points/plyutils.py -------------------------------------------------------------------------------- /models/pointops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/functions/pointops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/functions/pointops.py -------------------------------------------------------------------------------- /models/pointops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/setup.py -------------------------------------------------------------------------------- /models/pointops/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pointops/src/ballquery/ballquery_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/ballquery/ballquery_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/ballquery/ballquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/ballquery/ballquery_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/ballquery/ballquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/ballquery/ballquery_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/cuda_utils.h -------------------------------------------------------------------------------- /models/pointops/src/featuredistribute/featuredistribute_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/featuredistribute/featuredistribute_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/featuredistribute/featuredistribute_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/grouping/grouping_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/grouping/grouping_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/grouping/grouping_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/grouping/grouping_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/grouping/grouping_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/grouping/grouping_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/grouping_int/grouping_int_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/grouping_int/grouping_int_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/grouping_int/grouping_int_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/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/EasyRy/RepKPU/HEAD/models/pointops/src/grouping_int/grouping_int_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/interpolation/interpolation_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/interpolation/interpolation_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/interpolation/interpolation_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/interpolation/interpolation_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/interpolation/interpolation_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/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/EasyRy/RepKPU/HEAD/models/pointops/src/knnquery/knnquery_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/knnquery/knnquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/knnquery/knnquery_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/knnquery/knnquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/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/EasyRy/RepKPU/HEAD/models/pointops/src/knnquery_heap/knnquery_heap_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/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/EasyRy/RepKPU/HEAD/models/pointops/src/knnquery_heap/knnquery_heap_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/labelstat/labelstat_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/labelstat/labelstat_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/labelstat/labelstat_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/labelstat/labelstat_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/labelstat/labelstat_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/labelstat/labelstat_cuda_kernel.h -------------------------------------------------------------------------------- /models/pointops/src/pointops_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/pointops_api.cpp -------------------------------------------------------------------------------- /models/pointops/src/sampling/sampling_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/sampling/sampling_cuda.cpp -------------------------------------------------------------------------------- /models/pointops/src/sampling/sampling_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/sampling/sampling_cuda_kernel.cu -------------------------------------------------------------------------------- /models/pointops/src/sampling/sampling_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/pointops/src/sampling/sampling_cuda_kernel.h -------------------------------------------------------------------------------- /models/repkpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/repkpu.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/models/utils.py -------------------------------------------------------------------------------- /surf_recon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/surf_recon.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/train.py -------------------------------------------------------------------------------- /vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EasyRy/RepKPU/HEAD/vis.png --------------------------------------------------------------------------------