├── .gitignore ├── .pre-commit-config.yaml ├── .travis.yml ├── MANIFEST.in ├── README.rst ├── UNLICENSE ├── pointnet2 ├── __init__.py ├── _version.py ├── config │ ├── config.yaml │ ├── model │ │ ├── msg.yaml │ │ └── ssg.yaml │ ├── task │ │ ├── cls.yaml │ │ └── semseg.yaml │ └── task_model │ │ ├── cls-msg.yaml │ │ ├── cls-ssg.yaml │ │ ├── semseg-msg.yaml │ │ └── semseg-ssg.yaml ├── data │ ├── .gitignore │ ├── Indoor3DSemSegLoader.py │ ├── ModelNet40Loader.py │ ├── __init__.py │ └── data_utils.py ├── models │ ├── __init__.py │ ├── pointnet2_msg_cls.py │ ├── pointnet2_msg_sem.py │ ├── pointnet2_ssg_cls.py │ └── pointnet2_ssg_sem.py ├── train.py └── utils │ └── .gitignore ├── pointnet2_ops_lib ├── MANIFEST.in ├── pointnet2_ops │ ├── __init__.py │ ├── _ext-src │ │ ├── include │ │ │ ├── ball_query.h │ │ │ ├── cuda_utils.h │ │ │ ├── group_points.h │ │ │ ├── interpolate.h │ │ │ ├── sampling.h │ │ │ └── utils.h │ │ └── src │ │ │ ├── ball_query.cpp │ │ │ ├── ball_query_gpu.cu │ │ │ ├── bindings.cpp │ │ │ ├── group_points.cpp │ │ │ ├── group_points_gpu.cu │ │ │ ├── interpolate.cpp │ │ │ ├── interpolate_gpu.cu │ │ │ ├── sampling.cpp │ │ │ └── sampling_gpu.cu │ ├── _version.py │ ├── pointnet2_modules.py │ └── pointnet2_utils.py └── setup.py ├── pyproject.toml ├── requirements.txt ├── setup.py ├── tests ├── conftest.py ├── test_cls.py └── test_semseg.py └── tox.ini /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/.travis.yml -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | graft pointnet2/config/ 2 | -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/README.rst -------------------------------------------------------------------------------- /UNLICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/UNLICENSE -------------------------------------------------------------------------------- /pointnet2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/__init__.py -------------------------------------------------------------------------------- /pointnet2/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "3.0.0" 2 | -------------------------------------------------------------------------------- /pointnet2/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/config.yaml -------------------------------------------------------------------------------- /pointnet2/config/model/msg.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | use_xyz: True 3 | -------------------------------------------------------------------------------- /pointnet2/config/model/ssg.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | use_xyz: True 3 | -------------------------------------------------------------------------------- /pointnet2/config/task/cls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/task/cls.yaml -------------------------------------------------------------------------------- /pointnet2/config/task/semseg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/task/semseg.yaml -------------------------------------------------------------------------------- /pointnet2/config/task_model/cls-msg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/task_model/cls-msg.yaml -------------------------------------------------------------------------------- /pointnet2/config/task_model/cls-ssg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/task_model/cls-ssg.yaml -------------------------------------------------------------------------------- /pointnet2/config/task_model/semseg-msg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/task_model/semseg-msg.yaml -------------------------------------------------------------------------------- /pointnet2/config/task_model/semseg-ssg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/config/task_model/semseg-ssg.yaml -------------------------------------------------------------------------------- /pointnet2/data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/data/.gitignore -------------------------------------------------------------------------------- /pointnet2/data/Indoor3DSemSegLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/data/Indoor3DSemSegLoader.py -------------------------------------------------------------------------------- /pointnet2/data/ModelNet40Loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/data/ModelNet40Loader.py -------------------------------------------------------------------------------- /pointnet2/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/data/__init__.py -------------------------------------------------------------------------------- /pointnet2/data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/data/data_utils.py -------------------------------------------------------------------------------- /pointnet2/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/models/__init__.py -------------------------------------------------------------------------------- /pointnet2/models/pointnet2_msg_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/models/pointnet2_msg_cls.py -------------------------------------------------------------------------------- /pointnet2/models/pointnet2_msg_sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/models/pointnet2_msg_sem.py -------------------------------------------------------------------------------- /pointnet2/models/pointnet2_ssg_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/models/pointnet2_ssg_cls.py -------------------------------------------------------------------------------- /pointnet2/models/pointnet2_ssg_sem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/models/pointnet2_ssg_sem.py -------------------------------------------------------------------------------- /pointnet2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2/train.py -------------------------------------------------------------------------------- /pointnet2/utils/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | _ext 3 | -------------------------------------------------------------------------------- /pointnet2_ops_lib/MANIFEST.in: -------------------------------------------------------------------------------- 1 | graft pointnet2_ops/_ext-src 2 | -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/__init__.py -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/include/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/ball_query.h -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/cuda_utils.h -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/include/group_points.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/group_points.h -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/include/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/interpolate.h -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/include/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/sampling.h -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/utils.h -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/ball_query.cpp -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/bindings.cpp -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/group_points.cpp -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/interpolate.cpp -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/sampling.cpp -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_ext-src/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "3.0.0" 2 | -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py -------------------------------------------------------------------------------- /pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py -------------------------------------------------------------------------------- /pointnet2_ops_lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pointnet2_ops_lib/setup.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/setup.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/tests/test_cls.py -------------------------------------------------------------------------------- /tests/test_semseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/tests/test_semseg.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/erikwijmans/Pointnet2_PyTorch/HEAD/tox.ini --------------------------------------------------------------------------------