mirror of
https://github.com/wassname/Pointnet2_PyTorch.git
synced 2026-06-26 15:50:08 +08:00
erikwijmans
18 lines
537 B
ReStructuredText
18 lines
537 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
|
|
---------------------
|
|
|
|
- ``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
|