mirror of
https://github.com/wassname/Pointnet2_PyTorch.git
synced 2026-06-27 16:00:07 +08:00
erikwijmans
22 lines
611 B
ReStructuredText
22 lines
611 B
ReStructuredText
=================
|
|
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
|