├── .gitignore ├── README.md ├── cpp_wrappers ├── compile_wrappers.sh ├── cpp_neighbors │ ├── build.bat │ ├── build │ │ ├── lib.linux-x86_64-3.8 │ │ │ └── radius_neighbors.cpython-38-x86_64-linux-gnu.so │ │ ├── temp.linux-x86_64-3.8 │ │ │ ├── cpp_wrappers │ │ │ │ └── cpp_utils │ │ │ │ │ └── cloud │ │ │ │ │ └── cloud.o │ │ │ ├── neighbors │ │ │ │ └── neighbors.o │ │ │ └── wrapper.o │ │ └── temp.linux-x86_64-3.9 │ │ │ ├── cpp_wrappers │ │ │ └── cpp_utils │ │ │ │ └── cloud │ │ │ │ └── cloud.o │ │ │ ├── neighbors │ │ │ └── neighbors.o │ │ │ └── wrapper.o │ ├── neighbors │ │ ├── neighbors.cpp │ │ └── neighbors.h │ ├── radius_neighbors.cpython-38-x86_64-linux-gnu.so │ ├── radius_neighbors.cpython-39-x86_64-linux-gnu.so │ ├── setup.py │ └── wrapper.cpp ├── cpp_subsampling │ ├── build.bat │ ├── build │ │ ├── lib.linux-x86_64-3.8 │ │ │ └── grid_subsampling.cpython-38-x86_64-linux-gnu.so │ │ ├── temp.linux-x86_64-3.8 │ │ │ ├── cpp_wrappers │ │ │ │ └── cpp_utils │ │ │ │ │ └── cloud │ │ │ │ │ └── cloud.o │ │ │ ├── grid_subsampling │ │ │ │ └── grid_subsampling.o │ │ │ └── wrapper.o │ │ └── temp.linux-x86_64-3.9 │ │ │ ├── cpp_wrappers │ │ │ └── cpp_utils │ │ │ │ └── cloud │ │ │ │ └── cloud.o │ │ │ ├── grid_subsampling │ │ │ └── grid_subsampling.o │ │ │ └── wrapper.o │ ├── grid_subsampling.cpython-38-x86_64-linux-gnu.so │ ├── grid_subsampling.cpython-39-x86_64-linux-gnu.so │ ├── grid_subsampling │ │ ├── grid_subsampling.cpp │ │ └── grid_subsampling.h │ ├── setup.py │ └── wrapper.cpp └── cpp_utils │ ├── cloud │ ├── cloud.cpp │ └── cloud.h │ └── nanoflann │ └── nanoflann.hpp ├── data_utils ├── kpconv_loader.py ├── modelnet40_loader.py └── shapenet_loader.py ├── datasets └── ModelNet40.py ├── misc ├── layers.py ├── ops.py ├── pointconv_utils.py └── utils.py ├── networks ├── cls │ ├── blocks.py │ ├── data.txt │ ├── dgcnn.py │ ├── kernel_points.py │ ├── kernels │ │ └── dispositions │ │ │ └── k_015_center_3D.ply │ ├── kpconv.py │ ├── pointcnn.py │ ├── pointconv.py │ ├── pointnet.py │ └── pointnet2.py └── seg │ ├── dgcnn_partseg.py │ ├── pointcnn_partseg.py │ ├── pointconv_partseg.py │ ├── pointnet2_partseg.py │ └── pointnet_partseg.py ├── run_cls.sh ├── run_partseg.sh ├── train_cls.py └── train_partseg.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/README.md -------------------------------------------------------------------------------- /cpp_wrappers/compile_wrappers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/compile_wrappers.sh -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build.bat: -------------------------------------------------------------------------------- 1 | @echo off 2 | py setup.py build_ext --inplace 3 | 4 | 5 | pause -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/lib.linux-x86_64-3.8/radius_neighbors.cpython-38-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/lib.linux-x86_64-3.8/radius_neighbors.cpython-38-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.8/cpp_wrappers/cpp_utils/cloud/cloud.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.8/cpp_wrappers/cpp_utils/cloud/cloud.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.8/neighbors/neighbors.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.8/neighbors/neighbors.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.8/wrapper.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.8/wrapper.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.9/cpp_wrappers/cpp_utils/cloud/cloud.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.9/cpp_wrappers/cpp_utils/cloud/cloud.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.9/neighbors/neighbors.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.9/neighbors/neighbors.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.9/wrapper.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/build/temp.linux-x86_64-3.9/wrapper.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/neighbors/neighbors.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/neighbors/neighbors.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/neighbors/neighbors.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/neighbors/neighbors.h -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/radius_neighbors.cpython-38-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/radius_neighbors.cpython-38-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/radius_neighbors.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/radius_neighbors.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/setup.py -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/wrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_neighbors/wrapper.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build.bat: -------------------------------------------------------------------------------- 1 | @echo off 2 | py setup.py build_ext --inplace 3 | 4 | 5 | pause -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/lib.linux-x86_64-3.8/grid_subsampling.cpython-38-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/lib.linux-x86_64-3.8/grid_subsampling.cpython-38-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.8/cpp_wrappers/cpp_utils/cloud/cloud.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.8/cpp_wrappers/cpp_utils/cloud/cloud.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.8/grid_subsampling/grid_subsampling.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.8/grid_subsampling/grid_subsampling.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.8/wrapper.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.8/wrapper.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.9/cpp_wrappers/cpp_utils/cloud/cloud.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.9/cpp_wrappers/cpp_utils/cloud/cloud.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.9/grid_subsampling/grid_subsampling.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.9/grid_subsampling/grid_subsampling.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.9/wrapper.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/build/temp.linux-x86_64-3.9/wrapper.o -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/grid_subsampling.cpython-38-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/grid_subsampling.cpython-38-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/grid_subsampling.cpython-39-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/grid_subsampling.cpython-39-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.h -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/setup.py -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/wrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_subsampling/wrapper.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_utils/cloud/cloud.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_utils/cloud/cloud.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_utils/cloud/cloud.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_utils/cloud/cloud.h -------------------------------------------------------------------------------- /cpp_wrappers/cpp_utils/nanoflann/nanoflann.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/cpp_wrappers/cpp_utils/nanoflann/nanoflann.hpp -------------------------------------------------------------------------------- /data_utils/kpconv_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/data_utils/kpconv_loader.py -------------------------------------------------------------------------------- /data_utils/modelnet40_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/data_utils/modelnet40_loader.py -------------------------------------------------------------------------------- /data_utils/shapenet_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/data_utils/shapenet_loader.py -------------------------------------------------------------------------------- /datasets/ModelNet40.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/datasets/ModelNet40.py -------------------------------------------------------------------------------- /misc/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/misc/layers.py -------------------------------------------------------------------------------- /misc/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/misc/ops.py -------------------------------------------------------------------------------- /misc/pointconv_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/misc/pointconv_utils.py -------------------------------------------------------------------------------- /misc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/misc/utils.py -------------------------------------------------------------------------------- /networks/cls/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/blocks.py -------------------------------------------------------------------------------- /networks/cls/data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/data.txt -------------------------------------------------------------------------------- /networks/cls/dgcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/dgcnn.py -------------------------------------------------------------------------------- /networks/cls/kernel_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/kernel_points.py -------------------------------------------------------------------------------- /networks/cls/kernels/dispositions/k_015_center_3D.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/kernels/dispositions/k_015_center_3D.ply -------------------------------------------------------------------------------- /networks/cls/kpconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/kpconv.py -------------------------------------------------------------------------------- /networks/cls/pointcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/pointcnn.py -------------------------------------------------------------------------------- /networks/cls/pointconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/pointconv.py -------------------------------------------------------------------------------- /networks/cls/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/pointnet.py -------------------------------------------------------------------------------- /networks/cls/pointnet2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/cls/pointnet2.py -------------------------------------------------------------------------------- /networks/seg/dgcnn_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/seg/dgcnn_partseg.py -------------------------------------------------------------------------------- /networks/seg/pointcnn_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/seg/pointcnn_partseg.py -------------------------------------------------------------------------------- /networks/seg/pointconv_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/seg/pointconv_partseg.py -------------------------------------------------------------------------------- /networks/seg/pointnet2_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/seg/pointnet2_partseg.py -------------------------------------------------------------------------------- /networks/seg/pointnet_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/networks/seg/pointnet_partseg.py -------------------------------------------------------------------------------- /run_cls.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 python train_cls.py --model pointnet 2 | -------------------------------------------------------------------------------- /run_partseg.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 python train_partseg.py --model pointnet 2 | -------------------------------------------------------------------------------- /train_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/train_cls.py -------------------------------------------------------------------------------- /train_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jittor/PointCloudLib/HEAD/train_partseg.py --------------------------------------------------------------------------------