wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-08-12 11:40:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
erikwijmans
2017-12-26 18:43:17 -05:00
commit dc4e2b0db3
42 changed files with 3486 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.rst
+21
View File
@@ -0,0 +1,21 @@
=================
Pointnet2 PyTorch
=================
Partial implemention of `Pointnet2 <https://github.com/charlesq34/pointnet2>`_ written in `PyTorch <http://pytorch.org>`_.
The custom ops used by Pointnet2 are currently **ONLY** supported on the GPU using CUDA.
---------------------
Building CUDA kernels
---------------------
- ``cd utils``
- ``mkdir build && cd build``
- ``cmake .. && make``
------------------
Exampling training
------------------
Two training examples are provided by ``train_sem_seg.py`` and ``train_cls.py``. The datasets for both will be downloaded automatically by default