├── CMakeLists.txt ├── LICENSE ├── README.md ├── cfgs └── config_cls.yaml ├── cls └── model_cls_L6_iter_36567_acc_0.923825.pth ├── data ├── ModelNet40Loader.py ├── __init__.py └── data_utils.py ├── models ├── __init__.py └── densepoint_cls_L6_k24_g2.py ├── train_cls.py ├── train_cls.sh ├── utils ├── __init__.py ├── _ext │ ├── __init__.py │ └── pointnet2 │ │ └── __init__.py ├── build_ffi.py ├── cinclude │ ├── ball_query_gpu.h │ ├── ball_query_wrapper.h │ ├── cuda_utils.h │ ├── group_points_gpu.h │ ├── group_points_wrapper.h │ ├── interpolate_gpu.h │ ├── interpolate_wrapper.h │ ├── sampling_gpu.h │ └── sampling_wrapper.h ├── csrc │ ├── ball_query.c │ ├── ball_query_gpu.cu │ ├── group_points.c │ ├── group_points_gpu.cu │ ├── interpolate.c │ ├── interpolate_gpu.cu │ ├── sampling.c │ └── sampling_gpu.cu ├── linalg_utils.py ├── pointnet2_modules.py ├── pointnet2_utils.py └── pytorch_utils │ ├── __init__.py │ └── pytorch_utils.py └── voting_evaluate_cls.py /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/config_cls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/cfgs/config_cls.yaml -------------------------------------------------------------------------------- /cls/model_cls_L6_iter_36567_acc_0.923825.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/cls/model_cls_L6_iter_36567_acc_0.923825.pth -------------------------------------------------------------------------------- /data/ModelNet40Loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/data/ModelNet40Loader.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/data/data_utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/densepoint_cls_L6_k24_g2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/models/densepoint_cls_L6_k24_g2.py -------------------------------------------------------------------------------- /train_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/train_cls.py -------------------------------------------------------------------------------- /train_cls.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/train_cls.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/_ext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/_ext/pointnet2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/_ext/pointnet2/__init__.py -------------------------------------------------------------------------------- /utils/build_ffi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/build_ffi.py -------------------------------------------------------------------------------- /utils/cinclude/ball_query_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/ball_query_gpu.h -------------------------------------------------------------------------------- /utils/cinclude/ball_query_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/ball_query_wrapper.h -------------------------------------------------------------------------------- /utils/cinclude/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/cuda_utils.h -------------------------------------------------------------------------------- /utils/cinclude/group_points_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/group_points_gpu.h -------------------------------------------------------------------------------- /utils/cinclude/group_points_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/group_points_wrapper.h -------------------------------------------------------------------------------- /utils/cinclude/interpolate_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/interpolate_gpu.h -------------------------------------------------------------------------------- /utils/cinclude/interpolate_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/interpolate_wrapper.h -------------------------------------------------------------------------------- /utils/cinclude/sampling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/sampling_gpu.h -------------------------------------------------------------------------------- /utils/cinclude/sampling_wrapper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/cinclude/sampling_wrapper.h -------------------------------------------------------------------------------- /utils/csrc/ball_query.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/ball_query.c -------------------------------------------------------------------------------- /utils/csrc/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/ball_query_gpu.cu -------------------------------------------------------------------------------- /utils/csrc/group_points.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/group_points.c -------------------------------------------------------------------------------- /utils/csrc/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/group_points_gpu.cu -------------------------------------------------------------------------------- /utils/csrc/interpolate.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/interpolate.c -------------------------------------------------------------------------------- /utils/csrc/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/interpolate_gpu.cu -------------------------------------------------------------------------------- /utils/csrc/sampling.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/sampling.c -------------------------------------------------------------------------------- /utils/csrc/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/csrc/sampling_gpu.cu -------------------------------------------------------------------------------- /utils/linalg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/linalg_utils.py -------------------------------------------------------------------------------- /utils/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/pointnet2_modules.py -------------------------------------------------------------------------------- /utils/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/pointnet2_utils.py -------------------------------------------------------------------------------- /utils/pytorch_utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .pytorch_utils import * 2 | -------------------------------------------------------------------------------- /utils/pytorch_utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/utils/pytorch_utils/pytorch_utils.py -------------------------------------------------------------------------------- /voting_evaluate_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yochengliu/DensePoint/HEAD/voting_evaluate_cls.py --------------------------------------------------------------------------------