lucidrains/point-transformer-pytorch
A PyTorch implementation of the Point Transformer self-attention layer for 3D point cloud processing.

Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of the Point Transformer layer, based on a 2020 arXiv paper. The layer implements vector attention specifically designed for point cloud data, enabling tasks like classification and segmentation. It supports configurable hidden dimensions for position and attention MLPs, and optional k-nearest neighbor constraints to reduce computational cost by limiting attention to nearby points.