← all repositories

erikwijmans/Pointnet2_PyTorch

A PyTorch implementation of PointNet++ architectures for 3D point cloud processing with custom CUDA operations.

1.8k stars Python Computer VisionML Frameworks
Pointnet2_PyTorch
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of PointNet and PointNet++ neural network architectures for processing 3D point clouds. It includes custom CUDA operations for point cloud ops and supports multi-GPU training via DataParallel. The training pipeline is built with PyTorch Lightning and Hydra for experiment configuration. It supports classification and semantic segmentation tasks on 3D point cloud data.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.